[파이썬/알고리즘] 프로그래머스 - 전화번호 목록

[문제]

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

구조대 : 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):
    phone_book.sort()
    
    for i in range(len(phone_book) - 1):
        if phone_book[i+1].startswith(phone_book[i]):
            return False
    
    return True

 

프로그래머스에선 해시 문제로 분류되어 있지만, 사용하지 않고 다른 방법으로 구현하였다.

 

phone_book.sort()

먼저 정렬을 해준다.

정렬을 해주는 이유는 아래 예시를 보면 알 수 있다.

 

만약 phone_book이 ["119", "97674223", "1195524421"] 로 들어왔을 때 

정렬을 하게 되면, ['119', '1195524421', '97674223'] 이렇게 바뀐다.

 

접두어를 비교하기 더 용이해진다.

 

for i in range(len(phone_book) - 1):

     if phone_book[i+1].startswith(phone_book[i]):
             return False

반복문을 돌게 되는데 -1 을 해주는 이유는 모든 경우의 수를 따질 필요가 없기 때문이다.

문제에서 단 하나라도 접두어가 있으면 False를 반환하기 때문이고, 정렬을 하게 되면 접두어가 무조건 인접해 있을 것이기 때문이다. 

 

 ['119', '1195524421', '97674223']  이 예시에서도 정렬하니 첫 번째 119와 두 번째 119가 접두어로 겹치는 것을 볼 수 있다.

• i = 0일 때, phone_book[0]과 phone_book[1]을 비교한다. 즉, "119"와 "1195524421"을 비교한다.
• i = 1일 때, phone_book[1]과 phone_book[2]을 비교한다. 즉, "1195524421"과 "97674223"을 비교한다.

이처럼 반복문이 위 로직처럼 돌게 된다.

 

 

여기서 startswith(비교 변수)를 해주면 접두어가 비교된다.

끝나는 말이 들어가는지 확인하는 endswith 도 있다.