Python/알고리즘

프로그래머스-파이썬 (전화번호 목록)

SeongWon 2021. 6. 23. 15:26
반응형

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

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

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

 

 

<제한사항>

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

 

<입출력 예시>

 

 

 

<입출력 예시 설명>

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

 

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

 

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

 

<내 풀이>

 

def solution(phone_book):
    answer = True
    phone_book.sort()
    
    for i in range(len(phone_book)-1):
        a = len(phone_book[i]) 
        if phone_book[i] == phone_book[i+1][:a]:
            answer = False
            break
    
    return answer

1. 처음 phone_book을 정렬하고 시작하면 i번째와 i+1번째만 서로 비교하고 다음 인덱스로 넘어갈 수 있다.

  ex) phone_book = [212, 1234, 12] --> [12, 1234, 212] 로 정렬됨. 

 

2. i번 째의 번호가 i+1번 째의 "접두어"로 와야한다는건, 반드시 앞에 존재해야 하는 것이기 때문에, i번 번호의 길이를 a로 미리 정의해두고 i+1번 번호를 a만큼 인덱싱하여 비교한다.

 

3. 접두어로 존재하는 번호가 하나라도 있다면 false값을 반환하기 때문에, 작업 처리시간을 줄이기 위해서 false 값이 되는 순간 break로 반복문을 나오고 작업을 끝낸다.

 

 

<다른 풀이>

from itertools import combinations as c
def solution(phoneBook):
    answer = True
    sortedPB = sorted(phoneBook, key= len)
    for (a,b) in c( sortedPB,2):
        if a == b[:len(a)]:
            answer = False
    return answer

sorted(key=len) : 길이를 기준으로 정렬하면 시간 효율이 훨씬 좋다진다고 한다

반응형