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

문제 설명

---

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

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

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

제한 사항

---

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

입출력 예제

---

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

입출력 예 설명

---

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

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

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

solution.py

---

def solution(phone_book):
    answer = True
    order = sorted(phone_book)
    # order의 길이가 2일 때, 파이썬의 in을 사용하면 접두어가 아니더라도 포함한 것을 찾아 내기 때문에 접두어인지 아닌지 판별하기 위한 것 
    if len(order) == 2:
        if order[0][0] == order[1][0]:
            return False
        else:
            return answer
    # 정렬을 시키면 비슷한 숫자끼리 순서가 맞기 때문에 접두어로 시작하면 무조건 뒤에 배치하게 된다.
    for i in range(len(order)-1):
        if order[i] in order[i+1]:
            return False
        
    return answer

처음에 길이가 2짜리 예외를 찾지 못했다. 길이가 1인게 안들어가는 줄 알았는데, 그게 아니라 in의 특성상 그냥 접두어로 있는 것이 아닌 포함을 하고 있는것도 다 찾아내는 것이 문제였다. 그리고 길이가 3개짜리부터는 정렬이 되면 접두어로 시작하는 것이 아니면 알아서 사이에 다른 값이 들어가서 걸러지기 때문에 해당 부분에선 오류가 없다.
하지만 ["1", "31", "32"] 와 같은 배열에선 오류가 발생한다.

def solution(phone_book):
    answer = True
    order = sorted(phone_book)
    # zip을 통해서 order와 order[1]부터 나머지 값들을 for문안에 넣어준다.
    for i, j in zip(order, order[1:]):
    	# startswith라는 함수를 통해 str이 해당 값으로 시작하는 지 확인할 수 있다.
        if j.startswith(i):
            return False
        
    return answer