CodingTest/해시 알고리즘

프로그래머스 (Level 2) - 전화번호 목록

seongduck 2022. 8. 6. 16:26

1. 문제 설명

전화번호부에 적힌 전화번호 중, 한 번호가 다른 번호의 접두어인 경우가 있는지 확인하려 한다.

  • 다음과 같은 경우, 구조대 전화번호는 영석이의 전화번호의 접두사이다.
    • 구조대 : 119
    • 지영석 : 11 9552 4421
  • 전화번호부에 적힌 전화번호를 담은 배열 phone_book
  • 어떤 번호가 다른 번호의 접두어인 경우가 있으면 false, 그렇지 않은 true

2. 제한사항

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

3. 입출력 예

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

4. 풀이 접근

  1. 내림차순으로 정렬한다.
  2. i 번째와 i+1 번째를 비교한다.
  3. i+1 번째 수에 i번째가 속하면 false 출력 (b.startswith(a) 이용)

5. 코드

def solution(phone_book):
    phone_book = sorted(phone_book) #1
    
    for i in range(0, len(phone_book)-1): #2
        if (phone_book[i+1].startswith(phone_book[i])): #3
            return False
    return True
  • 이와 같은 코드를 간편하게 zip으로 묶어서 간결하게 나타낼 수도 있다.
def solution(phone_book):
    phone_book = sorted(phone_book) #1
    
    for a, b in zip(phone_book, phone_book[1: ]): #2
        if(b.startswith(a)): #3
            return False
    return True

6. 알아둘만한 문법

b.startswith(a)
  • a가 b 앞부분에 속하면 true 반환, 그렇지 않으면 false 반환
for a, b in zip(num, num[1: ])
  • num리스트 첫 번째 부분과 두 번째 부분을 각각 a, b로 선언하여 묶어주는 것