티스토리 뷰

728x90

문제

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

 

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

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

programmers.co.kr

 

 

풀이 및 소스코드

처음에는 sort() 후 걍 앞 뒤만 비교해서 앞의 숫자가 뒤에 숫자에 포함되어 있으면 false 리턴하도록 짰는데,

그렇게 되면

["113""44""4544"]

이 예시(테케 13번)을 틀리게 된다.

44는 4544의 접두어가 아니지만, 4544에 44가 포함되기때문에 false로 출력된다.

따라서 44의 문자열의 길이만큼만 비교하도록 수정했다.

def solution(phone_book):
    phone_book.sort()
    for i in range(len(phone_book)-1):
        if phone_book[i] in phone_book[i+1][:len(phone_book[i])]:
            return False
    return True
반응형