[프로그래머스 lv2] 전화번호 목록 (파이썬)

문제 설명

전화번호부에 적힌 전화번호 중, 한 번호가 다른 번호의 접두어인 경우가 있는지 확인하려 합니다.
전화번호가 다음과 같을 경우, 구조대 전화번호는 영석이의 전화번호의 접두사입니다.

• 구조대 : 119
• 박준영 : 97 674 223
• 지영석 : 11 9552 4421

전화번호부에 적힌 전화번호를 담은 배열 phone_book 이 solution 함수의 매개변수로 주어질 때, 어떤 번호가 다른 번호의 접두어인 경우가 있으면 false를 그렇지 않으면 true를 return 하도록 solution 함수를 작성해주세요.

제한 사항

• phone_book의 길이는 1 이상 1,000,000 이하입니다.
  • 각 전화번호의 길이는 1 이상 20 이하입니다.
  • 같은 전화번호가 중복해서 들어있지 않습니다.

입출력 예제

입출력 예제

입출력 예 설명

입출력 예 #1
앞에서 설명한 예와 같습니다.

입출력 예 #2
한 번호가 다른 번호의 접두사인 경우가 없으므로, 답은 true입니다.

입출력 예 #3
첫 번째 전화번호, “12”가 두 번째 전화번호 “123”의 접두사입니다. 따라서 답은 false입니다.

 

 

 

---

나의 풀이

첫 번째 시도

import re

def solution(phone_book):
    phone_book = sorted(phone_book, key=lambda x: (len(x), x), reverse=True)
    
    while len(phone_book):
        key = phone_book.pop()
        for num in phone_book:
            if re.match(f'^{key}', num): return False
    return True

우선 phone_book을 길이 순으로 정렬 후 길이가 같다면 숫자 크기 순으로 내림차순 정렬했다.

그러면 가장 작은 숫자가 리스트의 맨 오른쪽에 위치할 것인데, 리스트의 마지막 요소를 하나씩 빼서 비교한다.

가장 작은 숫자가 다른 숫자의 부분이 될 가능성이 가장 높다고 생각되어 이렇게 구성하였다.

 

phone_book에서 pop()을 하여 마지막 요소를 빼서 key 변수에 넣은 뒤

phone_book을 순서대로 탐색하며 key에 저장한 값으로 시작하는 숫자가 발견되면 False를 리턴하며 프로그램을 종료하도록 구성하였다.

(정규식을 사용하지 않고 if num[:len(key)] == key : return False 로 짜도 된다. )

 

그러나 이렇게 짰더니 효율성 테스트 3, 4에서 시간초과가 나왔다..

첫 번째 시도 정확성 테스트 결과

첫 번째 테스트 효율성 테스트 결과

---

두 번째 시도

import re

def solution(phone_book):
    phone_book = sorted(phone_book, key=lambda x: x, reverse=True)
    size = len(phone_book)
    while size > 1:
        key = phone_book.pop()
        size -= 1
        if re.match(f'^{key}', phone_book[-1]):
            return False
    return True

기존에는 문자열의 길이로 정렬을 하고

리스트의 끝에서 숫자를 하나 뽑아서 그 숫자를 나머지 전체 전화번호와 비교했다.

 

하지만 좀 더 생각해보니 문자열 기준으로 정렬을 하면 앞자리가 비슷한 것들끼리 묶일 것이다.

예를 들어 phone_book이

["12","123","1235","567","88"]라면, 정렬 이후

['88', '567', '1235', '123', '12'] 이렇게 될 것이다.

 

맨 오른쪽에 '12'가 존재하고, 바로 왼쪽을 보면 '12'로 시작하는 숫자들이 존재한다.

이렇게 바로 왼쪽의 숫자와 비교해서 접두어가 맞다면 바로 False를 리턴하면 된다.

 

테스트 결과를 보면 여전히 효율성 테스트 3, 4에서 시간 초과로 실패했지만

첫 번째 시도와 비교했을 때 테스트 20에서 3031.08ms 걸리던 알고리즘이 96ms로 눈에 띄게 줄어든 것을 확인할 수 있다.

두 번째 시도 정확성 테스트 결과

두 번째 테스트 효율성 테스트 결과

---

 

 

 

세 번째 시도(성공)

def solution(phone_book):
    phone_book = sorted(phone_book, key=lambda x: x, reverse=True) # 사전 순 내림차순 정렬
    idx = len(phone_book) - 1 # phone_book의 마지막 요소부터 검사
    while idx > 0:
        key = phone_book[idx] # 비교 대상
        if phone_book[idx-1][:len(key)] == key: return False # 바로 앞 숫자와 비교
        idx -= 1
        
    return True

시간을 더 줄이기 위해 쓸데없이 리스트에서 pop() 하는 부분을 없애고 인덱스로 조회하는 방식을 선택했다.

그리고 정규식도 사용하지 않고 바로 인덱스로 비교하니 시간이 엄청나게 빨라졌다.

 

세 번째 정확성 테스트 결과

 

테스트 20에서 첫 번째 시도에서는 3031.08ms, 두 번째 시도에서는 98ms가 소요되던 알고리즘이 1.88ms로 확연하게 줄어들었다.

세 번째 효율성 테스트 결과