54-9kyo-hwang

[9kyo-hwang]

🔗 문제 링크

16934 게임 닉네임

✔️ 소요된 시간

30분

✨ 수도 코드

1. 문제

$1 \le N \le 100,000$개의 문자열이 들어온다. 이 문자열은 알파벳 소문자로만 이루어져 있으며, 길이는 10을 넘지 않는다.
이 문자열들을 이용해 내부에 저장할 별칭을 만들고자 한다. 별칭은 다음과 같은 규칙으로 구성한다.

• 별칭은 문자열 접두사(Prefix) 중 가장 길이가 짧은 것을 사용한다. 이 때, 접두사가 이전에 등장한 문자열의 접두사가 아니어야 한다.
• 가능한 별칭이 없는 경우, 해당 문자열이 들어온 시점까지 들어온 같은 문자열 수 x를 계산해 이용한다. x가 1인 경우 문자열을 별칭으로 사용하고, 2 이상인 경우에는 문자열 뒤에 x를 붙여 별칭으로 사용한다.

예를 들어 $N = 5$이고 문자열들이 ["baekjoon", "startlink", "bakejoon", "beakjoon", "baekjoon"] 주어졌다면

• "baekjoon"의 별칭은 "b"가 된다.
• "startlink"의 별칭은 "s"가 된다.
• "bakejoon"의 별칭은 "bak"이 된다.
• "beakjoon"의 별칭은 "be"가 된다.
• "baekjoon"의 별칭은 "baekjoon2"가 된다.

N개의 문자열이 주어졌을 때, 들어온 순서대로 별칭을 한 줄에 하나씩 출력하라.

2. 설명

설명만 보면 Trie 자료구조를 사용하는 것이 가장 적절해보이긴 하나, N의 크기와 각 문자열의 길이가 크지 않아 단순히 Set/Map(Dictionary)을 사용해도 풀이가 가능하다.

우선 문자열이 주어지면 해당 문자열을 길이가 1, 2, 3, ...인 부분 문자열로 떼서 저장시켜야 한다.

• "baekjoon" 문자열이 들어왔다면 "b", "ba", "bae", "baek", ... 이런 식으로 저장한다.
• 그래야 뒤에 "bakejoon"에서 "b", "ba"가 중복되어 건너뛰어야 함을 인지할 수 있기 때문이다.

unordered_set<string> Storage;  // 별칭을 저장할 Set

int N; cin >> N;
while(N--)
{
    string Nickname; cin >> Nickname;
    string Prefix{};  // 접두사
    for(const char& ch : Nickname)
    {
        Prefix += ch;
        Storage.emplace(Prefix);  // 하나씩 덧붙이며 저장
    }
    ...
}

이 과정에서 만약 Storage에 저장되지 않은 별칭이 발견된다면 사용할 수 있음을 의미하므로 출력한다.

• "baekjoon"이 들어와 "b", "ba", "bae", "baek", ... 이 저장된 상태에서 "bakejoon"이 들어온 경우 "bak"가 Storage에 없다.
• 따라서 "bak"을 출력해야 한다.

bool IsPrinted = false;
string Prefix{};
for(const char& ch : Nickname)
{
    Prefix += ch;
    if(Storage.count(Prefix) == 0 && !IsPrinted)
    {
        cout << Prefix << "\n";
        IsPrinted= true;
    }
    Storage.emplace(Prefix);
}

• 위 예시를 다시 보면, 겹치지 않는 접두사는 "bak" 뿐만아니라 "bake", "bakej", ... 이 있다.
• 이 중 최초의 "가장 짧은" 접두사만 출력하도록 추가적인 bool 변수를 사용한다.

만약 저 반복문을 끝까지 돌았는데도 출력하지 못한 것이면 "중복된 문자열"이 들어왔음을 의미한다.
이 경우 "몇 개 들어왔는 지"를 알아야 하기 때문에 문자열을 Key로 해서 개수를 카운트 하는 Map을 이용한다.

unordered_map<string, int> NicknameMap;

int N; cin >> N;
while(N--)
{
    string Nickname; cin >> Nickname;
    NicknameMap[Nickname]++;  // 들어온 문자열의 개수 기록

    ...

    if(!IsPrinted)  // 앞선 반복문에서 적절한 별칭을 찾지 못했을 경우
    {
        string Prefix = Nickname;  // 입력된 문자열에
        if(NicknameMap[Nickname] > 1)
        {
            Prefix += NicknameMap[Nickname] + '0';  // 개수를 덧붙여서
        }
        Storage.emplace(Prefix);  // Storage에 저장하고
        cout << Prefix << "\n";  // 출력
    }
}

• 이 때 입력된 문자열 개수가 1보다 큰 경우를 기록하는 이유는 이전 반복문에서 문자열 중복에 의해 출력되지 못한 경우가 아닌 경우에도 이 조건문에 들어올 수 있기 때문이다.
• 예를 들어 "abcd" 문자열이 들어왔다면 별칭은 "a"가 되고 Storage에는 "a", "ab", "abc", "abcd"가 저장된다.
• 이후 "ab"라는 문자열이 들어왔다면, "ab"라는 문자열 자체는 이번이 최초이나 앞선 문자열에 의해 Storage에 "ab"라는 문자열이 존재한다.
• 즉 원래대로라면 "ab"가 별칭이 되어야 하나, IsPrint == false로 인해 별칭이 "ab1"으로 된다.
• 따라서 1을 붙이는 경우를 피하기 위해 저런 조건이 추가된다.

3. 전체 코드

#include <iostream>
#include <string>
#include <unordered_set>
#include <unordered_map>

using namespace std;

int main()
{
    cin.tie(nullptr)->sync_with_stdio(false);
    
    unordered_set<string> Storage;
    unordered_map<string, int> NicknameMap;
    
    int N; cin >> N;
    while(N--)
    {
        string Nickname; cin >> Nickname;
        NicknameMap[Nickname]++;
        
        bool IsPrinted = false;
        string Prefix{};
        for(const char& ch : Nickname)
        {
            Prefix += ch;
            if(Storage.count(Prefix) == 0 && !IsPrinted)
            {
                cout << Prefix << "\n";
                IsPrinted = true;
            }
            Storage.emplace(Prefix);
        }
        
        if(!IsPrinted)
        {
            string Prefix = Nickname;
            if(NicknameMap[Nickname] > 1)
            {
                Prefix += NicknameMap[Nickname] + '0';
            }
            Storage.emplace(Prefix);
            cout << Prefix << "\n";
        }
    }

    return 0;
}

📚 새롭게 알게된 내용

문제 이름이 재밌어 보여서 골랐는데 예전에 푼 5052 전화번호 목록 문제와 거의 유사해서 흠터레스팅...

[mjj111]

저는 Set/Map 사용해서 풀었는데 힘겹게 돌아가더군요 ㅋㅋㅋ ㅠ
이걸 30분만에 ㅎㄷ... 한 수 배워갑니다잉
문제 접근이 좀 까다로우면서 신박했습니다

import java.util.*;
import java.lang.*;
import java.io.*;

class Main {
    static private Set<String> nickPres;

    public static void main(String[] args) throws IOException {
        nickPres = new HashSet<>();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());

        Map<String, Integer> nameCount = new HashMap<>();
        for (int i = 0; i < N; i++) {
            String nowName = br.readLine();

            int nowCount = nameCount.getOrDefault(nowName, 0) + 1;
            nameCount.put(nowName, nowCount);

            System.out.println(getNickname(nowName, nameCount));
        }
    }

    static private String getNickname(String name, Map<String, Integer> nameCount) {
        int count = nameCount.get(name);
        if (count == 1) {
            String nicknamePres = getPres(name);
            return nicknamePres;
        }
        return name + count;
    }

    static private String getPres(String name) {
        boolean isNotApplied = false;
        String answerPre = "";
        
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < name.length(); i++) {
            sb.append(name.charAt(i));
            String nowPre = sb.toString();

            if (!nickPres.contains(nowPre) && !isNotApplied) {
                answerPre = nowPre;
                isNotApplied = true;
            }

            nickPres.add(nowPre);
        }

        return isNotApplied ? answerPre : name;
    }
}