[프로그래머스/python] 해시(hash) : 전화번호 목록

문제 설명

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

• 구조대 : 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):
    hash_map = {}
    for i in phone_book:
        hash_map[i] = 1

    for i in phone_book:
        temp = ""
        for number in i:
            temp += number
            if temp in hash_map and temp != i:
                return False
    return True

---

전화번호를 hash_map에 키로 저장한다.

전화번호를 한글자씩 늘리면서 temp에 저장하고

temp값이 hash_map에 있고 temp값이 전화번호 마지막 글자까지 가기 전이라면 False를 출력한다.

 

예를 들어 hash_map에 "00"과 "00232"가 있을 때 

00232를 한글자씩 temp에 저장하며 hash_map과 비교한다.

i = 00 일 경우,

temp = 0 일 때, hash_map에 "0"은 없다.

temp = 00 일 때, hash_map에 "00"이 있지만 i값인 00과 같다.

i = 00232 일 경우,

temp = 0 일 때, hash_map에 "0"은 없다.

temp = 00 일 때, hash_map에 "00"이 있고 i값은 00232이므로 00과 같지않다.

따라서 이 경우 False를 return한다.

 

모든 전화번호를 확인했음에도 return이 안됐다면 접두어인 경우가 없다는 뜻이므로 True를 return한다.