[백준] 5052번 전화번호 목록 (Python)

5052번: 전화번호 목록

문제

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

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

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

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

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

입력

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

출력

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

 

과정

배열을 정렬한 후 인접하고 있는 두 번호만 비교하면 일관성 없는 번호를 찾을 수 있다.

 

예제 입력

2
3
911
97625999
91125426
5
113
12340
123440
12345
98346

예제 출력

NO
YES

 

풀이

import sys
input = sys.stdin.readline

t = int(input().rstrip())

for _ in range(t):
    n = int(input().rstrip())
    nums = [input().rstrip() for _ in range(n)]
    nums.sort()
    answer = "YES"

    for n in range(len(nums)-1):
        if nums[n] == nums[n+1][:len(nums[n])]:
            answer = "NO"
            break

    print(answer)

 

한 줄씩 해석해보기

for _ in range(t):
    n = int(input().rstrip())
    nums = [input().rstrip() for _ in range(n)]
    nums.sort()
    answer = "YES"

• 입력받은 테스트 케이스 개수만큼 for문을 돌려준다.
• 개수 n을 입력받고 nums배열을 입력받는다.
• 배열을 정렬해준다.
• answer에 "YES"값을 넣어준다.

for n in range(len(nums)-1):
    if nums[n] == nums[n+1][:len(nums[n])]:
        answer = "NO"
        break

• 배열을 탐색한다.
• 만약 배열의 n번째 값이 그 다음값의 n번째 길이만큼 자른 값과 같다면
• answer에 NO를 넣어주고 break해준다

print(answer)

• 정답을 출력한다.