백준 전화번호 목록 파이썬(Python) 풀이

문제링크: https://www.acmicpc.net/problem/5052

문제

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

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

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

• 긴급전화: 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

풀이

• 트라이 이용
• 번호가 겹칠때마다 count 추가
• 번호가 일관성을 유지하면 True, 일관성을 유지하지 않으면 False
• 전화번호에서 숫자가 아닌 단어 입력은 무시 ex. "-"

예제 1: 911 입력

예제 1: 97625999

예제1: 단어 겹침 확인(911)

예제: 거꾸로 입력받을 때

 

소스코드

from sys import stdin
from collections import defaultdict


class TrieNode:
    def __init__(self):
        self.word = False
        self.cnt = 0  # 추가
        self.children = defaultdict(TrieNode)


class Trie:
    def __init__(self):
        self.root = TrieNode()
        self.consistency = True  # 추가

    # 단어 삽임
    def insert(self, word: str) -> None:
        if not self.consistency:    # 추가
            return

        node = self.root

        for char in word:
            if not char.isdigit():  # 숫자가 아닌 값 트라이에 넣지 않는다. ex) "123-456"
                continue
            node = node.children[char]
            node.cnt += 1

            if node.word:
                self.consistency = False
                return

        node.word = True  # 단어 마지막 true
        if node.cnt > 1:  # 거꾸로 일때
            self.consistency = False

    def check_consistency(self) -> bool:
        return self.consistency


def main():
    def input():
        return stdin.readline().rstrip()

    t = int(input())
    for _ in range(t):
        n = int(input())
        trie = Trie()
        for _ in range(n):
            tel = input()
            trie.insert(tel)
        print(['NO', 'YES'][trie.check_consistency()])


if __name__ == "__main__":
    main()