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

문제 설명

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

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

 

[python] 1차 시도..

def solution(phone_book):
    book_dict = {num : 0 for num in phone_book}
    for prefix in phone_book :
        for phone in book_dict.keys() :
            if len(prefix) > len(phone) :
                continue 
            elif prefix == phone[:len(prefix)] :
                book_dict[phone] += 1
    for i in book_dict.values() :
        if i > 1 :
            return False
    return True

 

흠... 이렇게 짰는데 알고리즘은 이상 없지만 효율성 테스트에서 실패가 뜬다.. ㅠ

 

아무래도 2중 for문을 바꿔야 할 것 같다..

 

def solution(phone_book):
    phone_book.sort(key=len,reverse=True)
    tuple_book = tuple(phone_book)
    for i, phone in enumerate(phone_book, 1) :
        if phone.startswith(tuple_book[i:]) :
            return False
    return True

 

흑.. 드디어 완성했다.. 해시로 풀진 않았지만.. 그래도 효율성 테스트에 통과를 했다.

 

[코드 설명]

phone_book.sort(key=len,reverse=True)

 

['1235', '123', '567', '12', '88']

 

우선 phone_book을 저렇게 정렬을 한다.. 그러면 자연스럽게 글자 길이가 긴 문자열이 맨 앞으로 오게 되는데

 

for i, phone in enumerate(phone_book, 1) :
    if phone.startswith(tuple_book[i:]) :
        return False

 

phone_book에서 문자열을 하나씩(긴 문자열 부터) 꺼내와서 튜플 안에 접두사가 있는지 확인을 한다. 

 

만약 접두사가 있다면 바로 return False를 한다.. 

 

+ phone_book을 tuple로 바꾼 이유는 startswith에는 튜플이나 문자열밖에 안 들어 가진다.

+ tuple_book[i:]를 한 이유는 현재 검사중인 문자열을 제외하고 검사하기 위해서 슬라이싱을 했다.