[알고리즘] BOJ 5052 전화번호 목록

[BOJ] 5052 전화번호 목록 바로가기

📍 문제

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

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

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

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

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

---

📍 입력

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

---

📍 출력

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

---

📍 풀이

✍ 처음 시도

• 처음에는 정규표현식과 re.match() 함수를 활용하여 문제를 해결하고자 하였다.
• re.match(lst, number) 에서 반환값이 None 이 아니라면 lst 에 '|number' 문자열을 추가하여 모든 접두어를 추가하는 방식으로 해결하고자 하였지만 시간초과가 발생하였다.
• sort() 함수가 숫자를 대상으로 하면 숫자의 값에 따라 정렬해 주지만 문자열을 대상으로 하면 사전순으로 정렬해 주는 것을 깨닫고 문제 해결 방법을 터득하였다.

✍ 문제 풀이

• 문자열을 대상으로 정렬 함수인 sort() 를 사용하면 사전순으로 정렬해준다.
• 사전순으로 정렬된 전화번호의 순서대로 접두어만 확인하여 같은 접두어를 가진 번호가 있다면 'NO' 를 반환하고 아니라면 'YES' 를 반환한다.

✍ 코드

from sys import stdin
# 한 번호가 다른 번호의 접두어인지 확인하는 함수
def check():
    for i in range(n-1):
        if numbers[i] == numbers[i+1][:len(numbers[i])]:
            return 'NO'
    return 'YES'
t = int(stdin.readline())
for _ in range(t):
    n = int(stdin.readline())
    numbers = sorted([stdin.readline().rstrip() for _ in range(n)]) # 번호 문자열을 사전순으로 정렬
    print(check())