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

문제 출처 : 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를 출력한다.

내 풀이

import sys
t = int(sys.stdin.readline())

def check(arr):
    for i in range(n - 1):
        if arr[i] == arr[i + 1][0 : len(arr[i])]:
            return "NO"
    return "YES"
    
for i in range(t):
    arr = []
    n = int(sys.stdin.readline())
   
    for j in range(n):
        arr.append(sys.stdin.readline().strip())
        
    arr.sort()
    
    print(check(arr))

우선 입력 받은 수 뒤에 개행이 들어가기 때문에 strip을 사용해 주어야 했는데 처음에 이를 사용하지 않아 오답이 났었다. 이 후에 2중 반복문을 이용해 해결을 하려 했으나 그렇게 해서 모든 경우를 탐색할 경우 시간초과가 나서 조금 애를 먹었다. 결과적으로 입력 받은 수를 정렬할 경우 바로 다음 수와 비교만 해주면 된다는 것을 알고 이중 반복을 없앴더니 해결이 되었다.