[ProblemSolving] 백준 - 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

풀이

문자열 문제다. 트라이로 풀지 않아서 쉽게 해결했다.

번호를 string으로 입력을 받고 정렬한다. [123, 234, 1234]는 [123, 1234, 234] 이런식으로 사전 정렬이 되고 여기서 이웃한 원소의 접두어만 비교하면 된다.

주의할 점은 번호를 입력받을 때, 개행 문자를 제거해줘야 한다.. 이부분에서 틀렸었다...

트라이는 나중에 도전하겠다..

코드

import sys
input = sys.stdin.readline
t = int(input())
for _ in range(t):
    n = int(input())
    numbers = []
    for _ in range(n):
        numbers.append(input().strip())
    numbers.sort()
    flag= True
    for i in range(n-1):
        if numbers[i] in numbers[i+1][:len(numbers[i])]:
            flag = False
            break
    if flag:
        print("YES")
    else:
        print("NO")