[백준] #5052 전화번호 목록(python)

수영·2022년 9월 28일

Python3 Trie 백준

백준

목록 보기

64/117

📌문제

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

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

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

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

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

입력

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

출력

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

예제 입력

예제 출력

NO
YES

백준 5052번 문제

💡Idea

문자열을 서로 비교하여 둘 중 하나의 문자열이 다른 하나의 문자열의 앞부분의 일부인가를 확인하는 문제입니다.

모든 문자열을 서로 비교하여 문제를 해결할 수도 있지만, 그렇게 되면 너무 많은 시간이 필요합니다.

따라서, 문자열을 효율적인 시간 내에 탐색할 수 있는 트라이를 사용하면 빠른 시간내에 문제를 해결할 수 있습니다!

💻코드

⏰ 시간 : 1236 ms / 메모리 : 51316 KB

import sys
input = sys.stdin.readline

class Node(object): #트라이의 노드 클래스
    def __init__(self, key, data=None):
        self.key = key
        self.data = data
        self.children = {}

class Trie(object): #트라이 클래스
    def __init__(self):
        self.head = Node(None)

	#문자열 삽입 함수로, 트라이의 어느 문자열이 string의 접두사면 False, 아니라면 True return
    def insert(self, string): 
        current = self.head
        for char in string:
            if char not in current.children:
                current.children[char] = Node(char)
            else:
            	#삽입하려는 문자열의 접두사가 존재하면 False return
                if current.children[char].data: return False
            current = current.children[char]
        current.data = string
        return True #접두사 존재하지 않으면 True return

t = int(input())
for _ in range(t):
    n = int(input())
    trie_tree = Trie() #트라이 생성
    #입력받은 전화번호 리스트를 길이 순으로 정렬
    phone_nums = sorted([input().strip() for __ in range(n)], key=lambda x: len(x))
    is_consist = True #접두사 존재 여부 저장 변수
    for num in phone_nums:
        if not trie_tree.insert(num): #접두사가 존재하는 경우(False가 return된 경우)
            is_consist = False 
            break
    print("YES" if is_consist else "NO")