파이썬(Python) 알고리즘 문제풀이 전화번호 목록 [프로그래머스 / Hash]

출처) 프로그래머스 Facebook

문제 설명

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

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

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

 

 

제한사항

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

 

 

입출력 예시

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

 

 

입출력 예 설명

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

 

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

 

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

 

 

풀이 코드

제한사항을 살펴보자.

1 <= len(phone_book) <= 10^6, 1 <= len(phone_book[i]) <= 20 이기 때문에 역시 일반적인 이중 반복문 완전 탐색으로 풀이하기는 힘들다.

 

처음에 생각한 풀이 방법은 아래와 같은데 구현 실력이 모자라서 그런건지 최대한 딕셔너리의 in 을 활용하려는 로직을 짤려고 해도 3번에서 계속 막혔다 ;;

1. 딕셔너리를 생성한다.

2. Key 는 전화번호, Value 는 False 로 갱신한다.

3. 하나의 전화번호를 기준으로 Value 값이 False 인 딕셔너리의 Key 중 Key.startswith(전화번호) 가 True 인지 확인한다.

    3-1. 여기서 자기 자신을 딕셔너리에서 찾아내면 안되니까 현재 선택된 전화번호를 Key 로 가지는 Value 값을 True 로 변경한 뒤 startswith() 를 사용해 판단한다.

4. 현재 선택된 전화번호를 접두어로 가지는 다른 전화번호가 있다면 바로 False 를 반환하고 함수를 종료한다.

 

그래서 생각한 다른 풀이 방법은 아래와 같다.

1. 딕셔너리를 생성한다.

2. Key 는 전화번호, Value 는 별 의미없는 아무 값으로 갱신한다.

3. 전화번호 리스트에서 하나의 전화번호를 선택하고 그 전화번호의 문자를 순회하도록 한다.

4. 미리 초기화한 접두어 변수를 갱신하고 이 접두어 변수가 딕셔너리에 있는지 파악한다.

5. 자기 자신은 탐색 대상에서 제외하도록 조건을 걸어주고, 탐색된다면 바로 False 를 반환하고 함수를 종료한다.

 

이 풀이 방법의 포인트는 하나의 전화번호를 접두어라고 가정해 다른 전화번호의 앞 글자를 찾아보는게 아니라, 하나의 전화번호에서 나올 수 있는 접두어를 만들어가면서 이 접두어가 딕셔너리에 있는지 파악하다는 것이다. 예를들어 ['119', '234', '567', '11'] 전화번호 리스트가 있다고 가정하면, '119' 자체를 접두어로 다른 '234', '567', '11' 을 비교하는게 아니라, '119' 가 가질 수 있는 '1', '11' 이라는 접두어가 딕셔너리에 담겨져 있는지 파악하는 식이다.

 

여기까지 코드는 아래와 같다.

pre += n 처럼 해도 상관없긴한데 문자열 + 문자열 연산은 굉장히 비효율적이니까 되도록이면 join() 을 적극 활용하자.

def solution(phone_book):
    phone_dict = dict()

    for number in phone_book:
        phone_dict[number] = 1

    # 전화번호 하나를 선택해서 자신이 가질 수 있는 접두어를 하니씩 만들어준다.
    # 접두어가 딕셔너리에 있다면 바로 False 반환
    for number in phone_book:
        pre = ''

        for n in number:
            # 접두어 만들기
            pre = ''.join([pre, n])

            # 만들어진 접두어가 딕셔너리에 있고, 그게 자신이 아닌 경우라면
            if pre in phone_dict and pre != number:
                return False

    return True

 

 

 

 

 

여기까지 코드는 아래와 같다.

''.join() 을 활용하려고 했는데, 완주하지 못한 선수는 단 한 명만 있다고하니 그냥 answer 에 넣고 반환하는 방법을 사용했고 Java 의 getOrDefault() 와 비슷하게 사용 가능한 dict[key] = dict.get(key, 0 ) + 1 를 사용했다.

def solution(participant, completion):
    p_dict = dict()

    # Key 값이 있다면 1 증가, 없다면 key 값 새로 생성
    for name in participant:
        p_dict[name] = p_dict.get(name, 0) + 1

    for name in completion:
        if name in p_dict:
            p_dict[name] -= 1

    for key, value in p_dict.items():
        if value > 0:
            answer = name

    return answer