[lv2] 전화번호 목록(Python 파이썬 풀이)

출처

https://programmers.co.kr/learn/courses/30/lessons/42577

코딩테스트 연습 - 전화번호 목록

문제 설명

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

구조대 : 119
박준영 : 97 674 223
지영석 : 11 9552 4421
전화번호부에 적힌 전화번호를 담은 배열 phone_book 이 solution 함수의 매개변수로 주어질 때, 어떤 번호가 다른 번호의 접두어인 경우가 있으면 false를 그렇지 않으면 true를 return 하도록 solution 함수를 작성해주세요.

제한 사항

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

입력 예제

  ["119", "97674223", "1195524421"]
  ["123","456","789"]
  ["12","123","1235","567","88"]

출력 예제

  false
  true
  false

입출력 예 설명

• 입출력 예 #1
  앞에서 설명한 예와 같습니다.
• 입출력 예 #2
  한 번호가 다른 번호의 접두사인 경우가 없으므로, 답은 true입니다.
• 입출력 예 #3
  첫 번째 전화번호, “12”가 두 번째 전화번호 “123”의 접두사입니다. 따라서 답은 false입니다.

풀이 및 해석

리스트 안에 최대 백 만개의 숫자들이 있으므로 단순히 for구문을 2개 돌린다면 시간 초과가 나올 것입니다. 그러므로 제약 조건을 추가해주도록 합니다. 만약 phone_book에 ["12", "123", "134"]가 있을 때를 가정해보겠습니다. "12"와 "123"을 비교하면 "123"안에 "12"가 있으므로 접두어를 사용했다는 것을 알 수 있지만, "134"에는 앞의 두 숫자와 비교를 해보아도 접두어로 지칭할 숫자가 없습니다. 비교를 할 때에는 자연스럽게 "12" 와 "134"중 "13"을 서로 비교하였을 것이고, "123"과 "134"를 비교할 때에는 크기 3을 모두 비교하는데에 사용하였을 것입니다. 또한, 오름차순으로 phone_book을 정렬한다면 for구문을 계속 돌지 않도록 break도 추가할 수 있습니다.

소스 코드

def solution(phone_book):
    # 오름차순 실행
    phone_book.sort()
    for i in range(len(phone_book) - 1):
        for j in range(i+1, len(phone_book)):
            # 만약 숫자가 더 크다면 더이상 볼 필요가 없으므로 break
            if int(phone_book[i]) < int(phone_book[j][:len(phone_book[i])]):
                break
            # phone_book은 오름차순이므로 한 쪽만 크기를 조절해주면 된다.
            elif phone_book[i] == phone_book[j][:len(phone_book[i])]:
                return False
    return True