[프로그래머스] 전화번호 목록 (hash lv.2) - Python

문제

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

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

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

제한 사항

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

---

잘못된 풀이(효율성 문제)

def solution(phone_book):
    for i in range(len(phone_book)):
        temp = phone_book[:]
        temp.pop(i)
        print(temp)
        for t in temp:
            print(t)
            if len(t) > len(phone_book[i]) and phone_book[i] == t[:len(phone_book[i])]:
                return False
    return True

for문이 두 번 돌아서 효율성이 떨어질 걸 알았지만.. 우선 생각나는 게 이거라 제출해봤는데 역시나 효율성 3, 4번에서 시간 초과가 떴다.

옳은 코드

구글링을 해서 sort를 한 다음 앞뒤 두 개씩 짝지어서 앞 번호가 뒷 번호의 접두어인지 확인하는 방법을 발견했다. 파이썬에 startswith라는 함수가 있다는 것도 처음 알았음...

from itertools import combinations

def solution(phone_book):
    phone_book.sort()
    for i in range(len(phone_book)-1):
        if phone_book[i+1].startswith(phone_book[i]):
            return False
    return True