백준 5052 전화번호 목록 (파이썬)

https://www.acmicpc.net/problem/5052

5052번: 전화번호 목록

처음에는 check 함수에서 이중 반복문을 써서 모든 수와 비교해야한다고 생각했다

다시 생각해보니 check 함수 전에 정렬을 하기 때문에 앞뒤 숫자만 비교해도 모든 경우를 확인할 수 있다

t = int(input())

def check():
    for i in range(len(a)-1):
        if a[i] == a[i+1][0:len(a[i])]:
            print('NO')
            return
    print('YES')

for _ in range(t):
    n = int(input())
    a = []
    for i in range(n):
        a.append(input().strip())
    a.sort()
    check()