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

문자열

 

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

 

첫번째 제출 (시간초과)

import re
def check():
    arr = []
    t = int(input())
    arr.append(input())
    for _ in range(t-1):
        s = input()
        for j in arr:
            if re.search(j, s):
                print("NO")
                return
        arr.append(s)

    print("YES")
    return

for _ in range(int(input())):
    check()

정규 표현식의 search를 이용해서 모든 문자을 확인하며 풀었지만 pypy3와 python3 모두 시간 초과가 발생하였습니다.

완전탐색 방법은 최대 50 * 10000 * 10000 *10 이라서 시간초과가 발생하는 것이었습니다.

 

두번째 풀이(정답, pypy3)

def check(arr):
    l = len(arr)
    for i in range(l-1):
        if arr[i+1].startswith(arr[i]):
            print("NO")
            return
    print("YES")
    return

for _ in range(int(input())):
    arr = []
    for _ in range(int(input())):
        arr.append(input())
    arr.sort()
    check(arr)

두번째 풀이는 먼저 모든 문자를 리스트에 저장하고 오름차순으로 정렬을 수행한 뒤, 현재 문자와 다음 문자를 startwith을 사용하여 비교합니다.

이렇게 해도 되는 이유는 다음과 같습니다,  

 

[1222, 114, 113, 3155, 63, 12311, 903]을 정렬할 경우 

 

1. 정수: [63, 113, 114, 903, 1143, 3155, 12311] 

2. 문자: [113, 114, 1143, 12311, 3155, 63, 903]

 

원소가 정수인 경우와 달리 문자인 경우는 앞에서부터 아스키코드 순으로 정렬을 수행하기 때문에 위와 같이 다음 문자만 비교해도 풀 수 있었습니다.