[백준 5052번] 파이썬 - 전화번호 목록

백준 5052 - 전화번호 목록

시간 제한 1초, 메모리 제한 256MB

# 조건

• 전화번호 목록이 주어진다.
  • 이때, 이 목록이 일관성이 있는지 없는지를 구하는 프로그램을 작성하시오.
• 전화번호 목록이 일관성을 유지하려면, 한 번호가 다른 번호의 접두어인 경우가 없어야 한다.
• 예를 들어, 전화번호 목록이 아래와 같은 경우를 생각해보자
  • 긴급전화: 911
  • 상근: 97 625 999
  • 선영: 91 12 54 26
• 이 경우에 선영이에게 전화를 걸 수 있는 방법이 없다.
• 전화기를 들고 선영이 번호의 처음 세 자리를 누르는 순간 바로 긴급전화가 걸리기 때문이다.
• 따라서, 이 목록은 일관성이 없는 목록이다.

입력

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

출력

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

# 접근 방법

• 트라이 구조를 이용하여 하나하나 비교해주어도 되지만, 접두어만 아니면 되기 때문에 조금 더 쉽게 구현해주었다.
• 주어진 단어를 먼저 words 리스트에 정렬한 후 sort를 활용하여 정렬해준다.
• 같은 단어로 시작하는 문자의 경우 사전순으로 정렬되기 때문에 결국은 바로 뒤의 단어만 확인해주면 된다.
• 현재 단어가 바로 뒤의 단어의 접두어라면 NO를 출력해주고 종료해주면 된다.

import sys  
sys.stdin = open('input.txt')  
input = sys.stdin.readline  

T = int(input())  
for _ in range(T):  
    N = int(input())  
    words = [input().strip() for _ in range(N)]  
    words.sort()  
    for i in range(N-1):  
        if words[i] == words[i+1][:len(words[i])]:  
            print('NO')  
            break  
    else:  
        print('YES')