[알고리즘] 백준 - 5052

[Gold IV] 전화번호 목록 - 5052

문제 링크

성능 요약

메모리: 295064 KB, 시간: 444 ms

분류

자료 구조, 정렬, 문자열, 트리, 트라이

제출 일자

2025년 4월 4일 20:20:26

문제 설명

전화번호 목록이 주어진다. 이때, 이 목록이 일관성이 있는지 없는지를 구하는 프로그램을 작성하시오.

전화번호 목록이 일관성을 유지하려면, 한 번호가 다른 번호의 접두어인 경우가 없어야 한다.

예를 들어, 전화번호 목록이 아래와 같은 경우를 생각해보자

• 긴급전화: 911
• 상근: 97 625 999
• 선영: 91 12 54 26

이 경우에 선영이에게 전화를 걸 수 있는 방법이 없다. 전화기를 들고 선영이 번호의 처음 세 자리를 누르는 순간 바로 긴급전화가 걸리기 때문이다. 따라서, 이 목록은 일관성이 없는 목록이다.

입력

첫째 줄에 테스트 케이스의 개수 t가 주어진다. (1 ≤ t ≤ 50) 각 테스트 케이스의 첫째 줄에는 전화번호의 수 n이 주어진다. (1 ≤ n ≤ 10000) 다음 n개의 줄에는 목록에 포함되어 있는 전화번호가 하나씩 주어진다. 전화번호의 길이는 길어야 10자리이며, 목록에 있는 두 전화번호가 같은 경우는 없다.

출력

각 테스트 케이스에 대해서, 일관성 있는 목록인 경우에는 YES, 아닌 경우에는 NO를 출력한다.

package baekjoon.april2025

fun main() {
    val t = readln().toInt()
    val ROOT = 1
    val MAX_VALUE = 10_000 * 11
    repeat(t) {
        val n = readln().toInt()
        var unused = 2
        var flag = false
        val trie = Array(MAX_VALUE) { IntArray(10) { -1 } }
        val check = BooleanArray(MAX_VALUE) { false }

        fun charToInt(c: Char): Int = c - '0'

        fun add(s: String): Boolean {
            var current = ROOT
            s.forEach {
                if (check[current]) return true
                if (trie[current][charToInt(it)] == -1) trie[current][charToInt(it)] = unused++
                current = trie[current][charToInt(it)]
            }
            if (trie[current].any { it != -1 }) return true
            check[current] = true
            return false
        }

        repeat(n) {
            val number = readln()
            if (add(number)) flag = true
        }

        if (flag) println("NO") else println("YES")
    }
}