[python] 프로그래머스 해시 전화번호 목록

📋 문제

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

구조대 : 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입니다.

알림

2021년 3월 4일, 테스트 케이스가 변경되었습니다. 이로 인해 이전에 통과하던 코드가 더 이상 통과하지 않을 수 있습니다.

🐍 이전 풀이(2021.06.14)

def solution(phone_book):
    answer = True
    
    phone_book.sort()
    
    for i in range(1, len(phone_book)):
        
        if phone_book[i][0] != phone_book[i - 1][0]:
            continue
        
        if len(phone_book[i]) == len(phone_book[i - 1]):
            continue
        elif len(phone_book[i]) > len(phone_book[i - 1]):
            if phone_book[i - 1] == phone_book[i][:len(phone_book[i - 1])]:
                return False
        elif len(phone_book[i]) < len(phone_book[i - 1]):
            if phone_book[i] == phone_book[i - 1][:len(phone_book[i])]:
                return False
    return answer

🐍 python 풀이

문제 예시에서 나온대로 "119"가 통째로 접두어에 포함해야 False이다. "119"를 접두어로 가지는 수의 예시는 "1191" 또는 "1191234" 등이 있다. 일부만 포함한 "11234"는 안된다는 의미이다.

파이썬의 리스트 정렬함수 sort()는

1. 문자 대소 판단
2. 문자 길이 판단

의 순서로 정렬하므로 예시 3번을 정렬하면 ["12", "123", "1235", "567", "88"]로 정리된다. 주어진 번호 리스트를 정렬함으로써 이중for문을 사용한 O(n**2)에서 바로 옆의 숫자만 비교해도 되므로 o(n)으로 줄일 수 있다.

주어진 번호를 오름차순으로 정렬 후 현재 번호의 바로 앞 숫자와 비교하여 바로 앞 숫자의 길이만큼 앞부분이 동일하면 False를 반환하였다.

def solution(phone_book):
    answer = True
    
    # 주어진 전화번호의 개수가 1일 경우 접두어가 동일할 경우가 없으므로 True 반환
    if len(phone_book) == 1:
        return answer
    
    phone_book.sort()
    
    for i in range(1, len(phone_book)):
        before_number = phone_book[i - 1]
        if before_number == phone_book[i][:len(before_number)]:
            return False
    
    return answer

python의 시간복잡도에 따르면 list의 slice는 O(k)의 시간복잡도를 가진다. k는 자르고자 하는 값의 크기이다. 즉 list의 slice는 자르고자하는 값의 인덱스를 찾으면 연산을 종료한다는 의미이다. 그럼 string의 slice도 유사하게 자르고자하는 문자의 인덱스를 찾게되면 연산을 종료한다는 것을 유추할 수 있다. 위 코드는 O(n)*O(k)의 시간복잡도를 가지는데 k는 n보다 작은 값이므로 최종 시간복잡도는 O(n)이 된다.

이전 코드를 보면 문자열의 길이를 비교 후 각 상황에 따라 문자열을 비교하였는데 애초에 파이썬의 정렬함수는 크기별, 길이별로 정렬하므로 필요하지 않은 과정이다.