[프로그래머스-Lv2] 전화번호 목록 / Python

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

import sys

input = sys.stdin.readline
phone_book = list(map(str, input().split()))


def solution(phone_book):
    phone_book.sort()
    for a, b in zip(phone_book, phone_book[1:]):
        if a == b[:len(a)]:
            return False
    return True


print(solution(phone_book))

 

해설

리스트를 정렬한다.

zip을 이용해 연속된 두 개의 데이터를 a, b로 치환해주고 b의 첫 a만큼의 길이가 a와 같다면 접두어에 포함이 되므로 false를 리턴한다.