알고리즘 문제/백준

전화번호 목록 #5052

BEstyle 2023. 1. 18. 05:01

문제

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

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

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

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

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

입력

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

출력

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

예제 입력 1 복사

2
3
911
97625999
91125426
5
113
12340
123440
12345
98346

예제 출력 1 복사

NO
YES

출처

ICPC > Regionals > Europe > Northwestern European Regional Contest > Nordic Collegiate Programming Contest > NCPC 2007 A번


# https://www.acmicpc.net/problem/

'''
1. 아이디어 :
    1) Trie로 풀 수 있는 문제이다. 먼저 hashmap에 전화번호를 한 자리씩 등록하면서, 중복이 되는지 확인한다.
2. 시간복잡도 :
    1) O(n) * O(1) * O(1) = O(n)
    - n개의 전화번호를 등록하는데, 각각의 전화번호를 등록하는데 O(1)이 걸리고, 중복이 되는지 확인하는데 O(1)이 걸린다.
3. 자료구조 :
    1) Trie
'''

import sys

input = sys.stdin.readline


class TrieNode:
    def __init__(self):
        self.children = {}
        self.endOfWord = False


class Trie:
    def __init__(self):
        self.root = TrieNode()


t = int(input())
for tests in range(t):
    Tries = Trie()
    n = int(input())
    ans = "YES"
    for nums in range(n):
        phone = input().rstrip()
        cur = Tries.root
        for i in range(len(phone)):
            if cur.endOfWord:
                ans = "NO"
                break
            if phone[i] in cur.children and i == len(phone) - 1:
                ans = "NO"
                break
            if phone[i] not in cur.children:
                cur.children[phone[i]] = TrieNode()
            cur = cur.children[phone[i]]
        cur.endOfWord = True
    print(ans)