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

5052번: 전화번호 목록

 

코드

import sys
input = sys.stdin.readline

for _ in range(int(input())):
    n = int(input())
    phones = [input().rstrip() for _ in range(n)]

    phones.sort()
    possible = True
    for i in range(len(phones)-1):
        if phones[i] == phones[i+1][:len(phones[i])]:
            possible = False
            break
    print("YES" if possible else "NO")

 

문자열 상태를 정렬하면 앞글자에서부터 비교하며 정렬하게 된다. 그러므로 전화번호 입력을 정수로 바꾸지 않고 문자열 그대로 받아서 정렬하면 같은 접두어를 가진 전화번호는 인접하게 된다. 이렇게 하면 한 번호를 전체와 비교해볼 필요 없이 맞붙어있는 두 번호만 비교해보면 된다.