[프로그래머스/Level2/파이썬3(python3)] 전화번호 목록

[프로그래머스/Level2/파이썬3(python3)] 전화번호 목록

문제

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

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

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

제한사항

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

입출력 예

phone_book	return
["119", "97674223", "1195524421"]	false
["123","456","789"]	true
["12","123","1235","567","88"]	false

입출력 예 설명

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

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

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

풀이

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

설명

내가 처음에 시도했던 코드는 이렇다.

def solution(phone_book):
    answer = True
    for i in range(len(phone_book)):
        for j in range(len(phone_book)):
            if i != j and phone_book[i] == phone_book[j][0:len(phone_book[i])]:
                answer = False
                break
    return answer

문제를 듣자마자 생각한 방법이었다. 특정 숫자가 들어오면 그 숫자를 포함하고 있는 다른 문자열을 발견하면 answer를 False로 하고 반복문을 멈추는 것. 정확도는 100%떴지만 효율성은 0%뜨더라. 아마 제한사항에서 배열의 갯수가 최대 100만 개라고 했으니 100만-1번째에 123, 100만번째에 123456789가 있는 등등 worst case의 경우 O(n^2)의 시간복잡도를 갖고 있기 때문인 것 같았다. 그러니까 최소 O(n^2)은 넘지 않는 시간복잡도를 가진 로직을 생각해야 했다. 그렇게 떠오른게 풀이 란의 로직이다.
 파일 정렬을 이름순으로 해본 사람들은 알겠지만 문자열을 정렬하면 같은 문자열을 포함하고 있는 애들이 바로 뒤에 배치된다. 예를 들어 ['123', '456', '789', '1234555']라는 phone_book이 주워졌으면 정렬했을 때 ['123', '1234555', '456', '789']가 된다. 그러니까 자기 바로 뒤에 있는 요소하고만 검사해보면 자기를 접두어로 포함하고 있는 요소가 있는지 검사할 수 있게 되는 것이다. 그렇게 해결하였다.