[백준] # 5052. 전화번호 목록 (파이썬)

 

문제 링크

 

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

5052번: 전화번호 목록

 

문제 내용

 

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

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

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

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

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

 

입력

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

출력

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

 

 

IDEA

 

트라이 알고리즘을 사용하면 문자열에서 문자를 하나씩 트리에 저장하며 비교 가능할 것 같아 사용했다.

문자열 저장이 끝나면 그 노드에 Data를 저장하므로, 전화번호목록을 리스트에 int로 저장한 후 정렬하고, 다시 string으로 저장해서 트라이 알고리즘을 사용하여 만약 노드에 Data가 존재하면 일관성이 없는 경우이다. 

 

CODE

class Node(object):
    def __init__(self, key, data=None):
        self.data = data
        self.children = {}

class Trie(object):
    def __init__(self):
        self.head = Node(None)
        self.consistency = "YES"
    # 전화번호(문자로 입력 받음) 저장
    def insert_and_check(self, phone_number):
        current_node = self.head   
        # 삽입할 전화번호의 각각의 숫자(문자)에 대해 자식 Node를 만들며 내려간다.
        for num in phone_number:
            # 자식 Node들 중 같은 숫자가 없으면 Node 새로 생성
            if num not in current_node.children:
                current_node.children[num] = Node(num)  # 딕셔너리 생성
                # 현재 노드에 data가 존재한다면, 일관성이 없는 것
                if current_node.data != None:
                    self.consistency = "NO"
            # 같은 문자가 있으면 노드를 따로 생성하지 않고, 해당 노드로 이동
            current_node = current_node.children[num]
            
        # 전화번호가 끝난 지점의 노드의 data값에 해당 전화번호를 입력
        current_node.data = phone_number
for _ in range(int(input())):
    n = int(input())
    phones = []
    for i in range(n):
        phones.append(input())  # 전화번호 문자열로 저장
    phones.sort() # 전화번호부 정렬
    tree = Trie()
    for phone_number in phones:
        tree.insert_and_check(phone_number)
        # 불필요한 시간소모 줄이기 위함.
        if tree.consistency == "NO":
            break
    print(tree.consistency)

 

사용 알고리즘

 

트라이 알고리즘

 

풀이 평

 

트라이 알고리즘에 대해 공부할 수 있어서 좋았다.