백준 5052 전화번호 목록 python

5052번: 전화번호 목록

문제

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

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

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

 

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

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

입력

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

출력

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

풀이

import sys
input = sys.stdin.readline

t = int(input())
# 테스트 케이스의 수를 받아줍니다.
for tc in range(t):
    n = int(input())
    # 확인할 번호의 수를 받아줍니다.
    numbers = [str(input().strip()) for _ in range(n)]
    numbers.sort()
    # 입력받는 전화번호를 문자열로 받아 리스트에 넣어주고
    # 번호가 담긴 리스트를 내림차순으로 정렬해줍니다.
    result = True
    # 일관성이 있다고 기본 형태를 잡아줍니다.
    for i in range(n - 1):
        # 연속한 두 번호를 비교할것이기 때문에 n - 1위치의 인덱스까지 확인이 가능합니다.
        length = len(numbers[i])
        if numbers[i] == numbers[i + 1][:length]:
            result = False
            break
            # numbers[i]의 길이가 length라고 할 때
            # numbers[i + 1]의 앞에서부터 length의 길이 문자열이 numbers[i]와 같다면
            # 연속하지 않으므로 표시해주고 반복을 멈춥니다.
    if result:
        print('YES')
    else:
        print('NO')
        # 연속하다면 'YES'아니면 'NO'를 출력해줍니다.