[Gold III] Title: 게임 닉네임, Time: 364 ms, Memory: 118008 KB -BaekjoonHub

Author: 051198Hz

백준/Gold/16934. 게임 닉네임/README.md

@@ -0,0 +1,40 @@
+# [Gold III] 게임 닉네임 - 16934 
+
+[문제 링크](https://www.acmicpc.net/problem/16934) 
+
+### 성능 요약
+
+메모리: 118008 KB, 시간: 364 ms
+
+### 분류
+
+자료 구조, 문자열, 트리, 집합과 맵, 해시를 사용한 집합과 맵, 트라이
+
+### 제출 일자
+
+2025년 9월 10일 13:09:10
+
+### 문제 설명
+
+<p>스타트링크에서 매우 재미있는 게임을 만들었다. 이 게임은 정말 재미있다.</p>
+
+<p>게임에는 유저가 접속하는 기능이 있고, 각 유저는 가입할 때, 자신의 닉네임을 정해야 한다. 닉네임은 알파벳 소문자로만 이루어져 있고, 두 유저가 같은 닉네임을 정하는 것도 가능하다.</p>
+
+<p>이 게임은 유저의 닉네임을 이용해서 내부에 저장할 별칭을 만든다. 별칭은 유저에게 보여지지는 않고, 내부에서만 사용된다. 저장 공간을 최소로 하기 위해서 별칭의 길이를 최소로 하려고 한다.</p>
+
+<p>별칭은 유저 닉네임의 접두사(Prefix) 중에서 가장 길이가 짧은 것을 사용한다. 이때, 접두사가 이전에 가입한 닉네임의 접두사가 아니어야 한다. 가능한 별칭이 없는 경우에는 유저가 가입한 시점까지 같은 닉네임으로 가입한 사람의 수 x를 계산해야 한다. x가 1인 경우에는 닉네임을 별칭으로 사용하고, x가 2 이상인 경우에는 닉네임의 뒤에 x를 붙여서 별칭으로 사용한다.</p>
+
+<p>예를 들어, 닉네임을 "baekjoon"으로 정한 유저가 가입하면, 이 유저의 별칭은 "b"가 된다. </p>
+
+<p>그 다음, 닉네임이 "startlink"로 정한 유저가 가입하면, 이 유저의 별칭은 "s"이다. "bakejoon"이 닉네임인 유저가 가입하면, 별칭은 "bak"가 되고, "beakjoon"인 유저가 가입하면, 별칭은 "be"가 된다. 마지막으로 "baekjoon"으로 유저가 가입하면 별칭은 "baekjoon2"가 된다.</p>
+
+<p>유저가 가입한 순서대로 닉네임이 주어졌을 때, 각 유저의 별칭을 구해보자. 위의 규칙을 이용해 별칭을 정하면 두 유저가 같은 별칭을 갖는 것도 가능하다.</p>
+
+### 입력 
+
+ <p>첫째 줄에 가입한 유저의 수 N(1 ≤ N ≤ 100,000)이 주어진다. 둘째 줄부터 N개의 줄에는 유저의 닉네임이 가입한 순서대로 한 줄에 하나씩 주어진다. 닉네임은 알파벳 소문자로만 이루어져 있고, 길이는 10을 넘지 않는다.</p>
+
+### 출력 
+
+ <p>유저가 가입한 순서대로 별칭을 한 줄에 하나씩 출력한다.</p>
+

백준/Gold/16934. 게임 닉네임/게임 닉네임.swift

@@ -0,0 +1,90 @@
+final class Trie<T: BidirectionalCollection> where T.Element: Hashable {
+    class Node {
+        var childList = [T.Element: Node]()
+        var endOfWord: Bool
+
+        init(_ _endOfWord: Bool) {
+            endOfWord = _endOfWord
+        }
+
+        func hasChild(for c: T.Element) -> Node? {
+            return childList[c]
+        }
+        func addChild(for key: T.Element) -> Node {
+            let child = Node(false)
+            childList.updateValue(child, forKey: key)
+            return child
+        }
+    }
+
+    let root = Node(false)
+
+    func insert(_ word: T) -> [T.Element] {
+        var prefix = [T.Element]()
+        var isShortestPrefix = true
+        // 단어를 입력받는다.
+        // 단어의 알파벳을 순회하며
+        // 루트노드 부터 해당 알파벳이 존재한다면
+        // 그냥 쭉쭉 그 노드로 넘기고 현재 탐색 노드를 그 노드의 해당 자식 노드로 교체
+        // 존재하지 않는다면, 탐색중인 노드에 새 자식 노드를 등록하고 해당 자식 노드로 교체
+        var currentNode = root
+        for c in word {
+            if let child = currentNode.hasChild(for: c) {
+                prefix.append(c)
+                currentNode = child
+                continue
+            }
+            currentNode = currentNode.addChild(for: c)
+            guard isShortestPrefix else { continue }
+            prefix.append(c)
+            isShortestPrefix = false
+        }
+        currentNode.endOfWord = true
+
+        return prefix
+    }
+
+    func contains(_ target: T) -> Bool {
+        var currentNode = root
+        for element in target {
+            guard let nextNode = currentNode.hasChild(for: element) else { return false }
+            currentNode = nextNode
+        }
+        return currentNode.endOfWord
+    }
+
+    @discardableResult
+    func delete(_ target: T) -> Bool {
+        var currentNode = root
+        // 삭제할 대상의 마지막 문자열 노드까지 내려감
+        for element in target {
+            guard let nextNode = currentNode.hasChild(for: element) else { return false }
+            currentNode = nextNode
+        }
+        // 마지막 노드까지 내려갔는데, 대상의 끝이 아니라면 삭제 불가
+        guard currentNode.endOfWord else { return false }
+        currentNode.endOfWord = false
+        return true
+    }
+}
+
+
+let n = Int(readLine()!)!
+let trie = Trie<String>()
+var wordCountMap = [String: Int]()
+
+var answer = ""
+
+for _ in 0..<n {
+    let word = readLine()!
+    let nickName = trie.insert(word)
+    if let count = wordCountMap[word] {
+        wordCountMap[word]! += 1
+        answer.write(word + "\(count+1)\n")
+    } else {
+        wordCountMap[word] = 1
+        answer.write(String(nickName)+"\n")
+    }
+}
+
+print(answer)