문제

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

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

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

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

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

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

입력

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

출력

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

아이디어

✅시간 초과

1. 데이터를 받아서 길이순으로 정렬한다.
2. 전화번호 개수만큼 반복하여 비교할 전화번호와 같은 길이로 자른다.
3. 중복을 제거했을때, 리스트의 길이가 처음과 달라지면 중복된 번호가 있으므로 'NO'를 출력하고, 중복이 없으면 'YES'를 출력한다.

✅clone

1. 리스트로 받아서 정렬시킨다.
2. 정렬을 했기때문에 바로 다음 데이터만 비교하면 된다.
3. 비교할 데이터와 길이를 맞춘 후 같으면 'NO'를 출력하고 아니면 'YES'를 출력한다.

내 코드(Python)_Error_시간 초과

import copy
import sys
input = lambda : sys.stdin.readline()

num = int(input())

for _ in range(num):
    phone = int(input())
    data = []
    for _ in range(phone):
        data.append(list(input().strip('\n')))
    data.sort(key= lambda x:(len(x)))
    for i in range(phone-1):
        data2 = copy.deepcopy(data)
        if len(data2[i]) < 10:
            for j in range(i+1, phone):
                data2[j] = data2[j][0:len(data2[i])]
        data2 = list(set(map(tuple,data2)))
        if len(data2) != phone:
            print('NO')
        else:
            print('YES')

다른 코드(Python)_Clone

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()

출처: https://chldkato.tistory.com/83