[백준] 5052. 전화번호 목록 - Gold 4

방근호·2023년 3월 14일
0

알고리즘

목록 보기
14/14

문제

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

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

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

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

입력

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

출력

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

풀이

<처음에 생각했던 것>
1. 문자열을 오름차순으로 정렬 후
2. 기준 문자열을 잡고 다음 문자열의 처음부터 해당하는 길이까지 비교했을 때 같으면, valid = False
3. 테스트 케이스는 맞으나.. 시간초과 발생

<트라이 자료구조 사용>
트라이에 대해 공부 후 사용

코드는 위 블로그를 참고하였으며, 문제에 맞게 커스터 마이징 하였음.
검색하는 과정에서 data를 가진 부분이 발견되는 경우에, 기존 문자열이 존재하는 것 문제 해결

코드

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)

    # 문자열 삽입
    def insert(self, string):
        curr_node = self.head

        # 삽입할 String 각각의 문자에 대해 자식Node를 만들며 내려간다.
        for char in string:
            # 자식Node들 중 같은 문자가 없으면 Node 새로 생성
            if char not in curr_node.children:
                curr_node.children[char] = Node(char)

            # 같음 문자가 있으면 노드를 따로 생성하지 않고, 해당 노드로 이동
            curr_node = curr_node.children[char]

        # 문자열이 끝난 지점의 노드의 data값에 해당 문자열을 표시
        curr_node.data = string

    # 문자열이 존재하는지 탐색!
    def search(self, string):
        # 가장 아래에 있는 노드에서부터 탐색 시작한다.
        curr_node = self.head

        for char in string:
            if curr_node.data is not None:
                return True

            if char in curr_node.children:
                curr_node = curr_node.children[char]
            else:
                return False

        # 탐색이 끝난 후에 해당 노드의 data값이 존재한다면
        # 문자가 포함되어있다는 뜻이다!
        if curr_node.data is not None:
            return True


for _ in range(int(input())):
    n = int(input())
    arr = [input() for _ in range(n)]
    arr.sort()
    valid = True

    t = Trie()
    t.insert(arr[0])
    for i in range(1, n):
        if t.search(arr[i]):
            valid = False
            break
        else:
            t.insert(arr[i])


    print('YES' if valid else 'NO')
profile
잘 설득하는 개발자가 되기 위해 노력합니다.

0개의 댓글