문제 설명

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

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

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

제한사항

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

문제 풀이

리스트 요소값을 비교하여 특정 값이 다른 값의 접두어가 되는 경우를 판단하는 문제이다.
쉽게 접근하기 위해서 먼저 리스트 내의 값들을 정렬했다.

정렬 후에는 이중 for문을 돌려 값을 서로 비교하고 포함관계에 있다면 False를, 그렇지 않으면 True를 반환하는 함수를 구현했다.

첫 번째 시도

def solution(phone_book):
    phone_book.sort()    # 리스트 정렬
    for i in range(len(phone_book)):
        for j in range(i+1, len(phone_book)):
            if phone_book[i] in phone_book[j]:
                return False
    return True

간단하게 phone_book 내의 리스트값과 바로 뒤의 리스트 값을 비교하여 포함관계에 있으면 False를, 그렇지 않으면 True를 반환하는 함수를 구현했다.

실행 결과

채점 결과

이중 for문을 구현했기 때문에 효율성 부분에서 실패할 것이라고 생각했는데, 그 이전에 정확도조차 통과하지 못했다.
그 이유는

if phone_book[i] in phone_book[j]

이 구문 때문이었다.
접두어 관계인 경우만 처리해야 하는데 (a in b)로 처리할 경우 접두어가 아닌 경우까지 배제시켜 버린 것이 원인이었다.
예를 들면 테스트 케이스가 ['119', '46589', '5311965'] 인 경우, 접두어에 해당하는 경우는 없음에도 False값을 출력하는 오류가 발생했다.

두 번째 시도

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

위의 잘못된 결과값을 내지 않기 위해 startswith 함수를 사용했다.

채점 결과

아니나 다를까 정확도 테스트는 통과했지만 효율성 테스트를 통과하지 못했다.
이중 for문을 구현했기 때문에 리스트 내의 값이 많아지는 경우 시간 초과가 발생하는 것으로 판단했다.

세 번째 시도

def solution(phone_book):
    phone_book.sort()
    for i, j in zip(phone_book, phone_book[1:]):
        if j.startswith(i):
            return False
    return True

이중 for문이 아닌 zip 내장함수를 사용했다.

두 그룹으로 데이터를 나눠 각각 i(전체 phone_book 리스트)와 j(phone_book[1:] 리스트) 변수에 할당하여 구현했다.

채점 결과