[코딩 테스트 연습(파이썬/Python)] 백준 5052번 _ 전화번호 목록

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

5052번: 전화번호 목록

[문제]

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

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

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

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

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

 

[입력]

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

[출력]

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

 

[아이디어]

1) 전화번호 목록을 정렬하면 연속된 문자열이 겹치는지 확인하는 것 만으로 일관성이 있는지 여부를 확인할 수 있다.

2) sys.stdin.readline으로 읽어오면 개행 문자가 포함된다,,^

3) zip()으로 전화번호 목록을 한 칸 민 두 개의 배열을 사용한다면, 매번 길이를 구하고 슬라이싱을 하지 않아도 해결이 가능하다.

import sys
input=sys.stdin.readline

t = int(input())
for _ in range(t):
    n = int(input())
    p = [input().rstrip() for _ in range(n)]
    p.sort()
    cnt = 1
    for i,x in enumerate(p[:-1]):
        tmp = len(x)
        if p[i+1][:tmp]==x:
            print('NO')
            cnt -=1
            break
    if cnt:
        print('YES')

개행 문자 제거하는 걸 깜빡해서 하루 종일 맞왜틀,,,,,,, 반복적인 길이 체크 + 슬라이싱 등으로 효율이 떨어지는 면이 있다.

import sys
input=sys.stdin.readline

t = int(input())
for _ in range(t):
    n = int(input())
    p = [input().rstrip() for _ in range(n)]
    p.sort()
    cnt = True
    for x,y in zip(p[:-1],p[1:]):
        if y.startswith(x):
            print('NO')
            cnt = False
            break
    if cnt:
        print('YES')

아주 좋은 방법인데, 테케가 충분하지 못한건지 혹은 배열 두 개를 slicing하고 zip하는게 생각보다 오래걸리는건지 모르겠지만 막상 시간에서 크게 차이가 나진 않았다. 트라이(trie) 자료구조를 활용해서 풀 수도 있다고 하는데, 다음에 공부해봐야겠다.