[Gold IV] Title: 전화번호 목록, Time: 876 ms, Memory: 77820 KB -BaekjoonHub

Author: 051198Hz

백준/Gold/5052. 전화번호 목록/README.md

@@ -0,0 +1,40 @@
+# [Gold IV] 전화번호 목록 - 5052 
+
+[문제 링크](https://www.acmicpc.net/problem/5052) 
+
+### 성능 요약
+
+메모리: 77820 KB, 시간: 876 ms
+
+### 분류
+
+자료 구조, 문자열, 정렬, 트리, 집합과 맵, 트라이
+
+### 제출 일자
+
+2025년 9월 10일 13:50:56
+
+### 문제 설명
+
+<p>전화번호 목록이 주어진다. 이때, 이 목록이 일관성이 있는지 없는지를 구하는 프로그램을 작성하시오.</p>
+
+<p>전화번호 목록이 일관성을 유지하려면, 한 번호가 다른 번호의 접두어인 경우가 없어야 한다.</p>
+
+<p>예를 들어, 전화번호 목록이 아래와 같은 경우를 생각해보자</p>
+
+<ul>
+	<li>긴급전화: 911</li>
+	<li>상근: 97 625 999</li>
+	<li>선영: 91 12 54 26</li>
+</ul>
+
+<p>이 경우에 선영이에게 전화를 걸 수 있는 방법이 없다. 전화기를 들고 선영이 번호의 처음 세 자리를 누르는 순간 바로 긴급전화가 걸리기 때문이다. 따라서, 이 목록은 일관성이 없는 목록이다. </p>
+
+### 입력 
+
+ <p>첫째 줄에 테스트 케이스의 개수 t가 주어진다. (1 ≤ t ≤ 50) 각 테스트 케이스의 첫째 줄에는 전화번호의 수 n이 주어진다. (1 ≤ n ≤ 10000) 다음 n개의 줄에는 목록에 포함되어 있는 전화번호가 하나씩 주어진다. 전화번호의 길이는 길어야 10자리이며, 목록에 있는 두 전화번호가 같은 경우는 없다.</p>
+
+### 출력 
+
+ <p>각 테스트 케이스에 대해서, 일관성 있는 목록인 경우에는 YES, 아닌 경우에는 NO를 출력한다.</p>
+

백준/Gold/5052. 전화번호 목록/전화번호 목록.swift

@@ -0,0 +1,96 @@
+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) -> Bool {
+        // 단어를 입력받는다.
+        // 단어의 알파벳을 순회하며
+        // 루트노드 부터 해당 알파벳이 존재한다면
+        // 그냥 쭉쭉 그 노드로 넘기고 현재 탐색 노드를 그 노드의 해당 자식 노드로 교체
+        // 존재하지 않는다면, 탐색중인 노드에 새 자식 노드를 등록하고 해당 자식 노드로 교체
+        var currentNode = root
+        for c in word {
+            if let child = currentNode.hasChild(for: c) {
+                currentNode = child
+                if currentNode.endOfWord { return false }
+                continue
+            }
+            currentNode = currentNode.addChild(for: c)
+        }
+        currentNode.endOfWord = true
+        guard currentNode.childList.isEmpty else { return false }
+        return true
+    }
+
+    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
+    }
+
+    func hasWordSamePrefix(from target: T) -> Bool {
+        var currentNode = root
+        var depth = 1
+        for element in target {
+            guard let nextNode = currentNode.hasChild(for: element) else { return false }
+            if nextNode.endOfWord, depth != target.count {
+                return true
+            }
+            currentNode = nextNode
+            depth += 1
+        }
+        return false
+    }
+
+    @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 t = Int(readLine()!)!
+
+outer: for _ in 0..<t {
+    let trie = Trie<String>()
+    // let n = Int(readLine()!)!
+    if let input = readLine(), let n = Int(input) {
+        var isValid = true
+        for _ in 0..<n {
+            let line = readLine()!
+            if !isValid { continue }
+            if !trie.insert(line) {
+                isValid = false
+            }
+        }
+        print(isValid ? "YES" : "NO")
+    }
+}