[Programmers] 전화번호 목록 (Python3)

문제 설명

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

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

입출력 예 #2

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

def solution(phone_book):
    phone_book = sorted(phone_book)
    
    for i in range(len(phone_book) - 1):
        print(i)
        if phone_book[i] in phone_book[i + 1][0 : len(phone_book[i])]:
            return False
    return True
        
    
        
        
        

solution(["1","231","789"])

0
1





True

문제풀이

이 문제는 얼핏보면 for문 두개 돌려서 풀면 될것처럼 보이지만 코테가 다 그렇듯이 for문 두개를 돌리면 무조건 시간초과가 난다. 그래서 어차피 전화번호가 다른 전화번호에 포함된 경우가 아닌 접두어인지를 묻는 문제이기 때문에, 더 긴 번호가 짧은 번호의 접두어가 될 수 없다. 따라서 정렬을 통해 접두어의 가능성이 있는 번호들을 모아주고, 이를 다음 하나의 번호에만 체크해봄으로써 반복문의 횟수를 줄여 해결할 수 있다.