programmers lv2 전화번호 목록 python

programmers lv2 전화번호 목록 python

문제 설명 : 해시(hash)

전화번호부에 적힌 전화번호 중, 한 번호가 다른 번호의 접두어인 경우가 있는지 확인하려 합니다.

전화번호가 다음과 같을 경우, 구조대 전화번호는 영석이의 전화번호의 접두사입니다.

구조대 : 119

박준영 : 97 674 223

지영석 : 11 9552 4421

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

제한 사항

phone_book의 길이는 1 이상 1,000,000 이하입니다.

각 전화번호의 길이는 1 이상 20 이하입니다.

첫 번째 문제 풀이(시간 초과)

[1] 정렬을 먼저합니다.

[2] 2중 for문을 돌면서 find()를 이용해 문자열이 있는지 확인합니다.

def solution(phone_book):

    answer = True

    # 정렬을 먼저한다.

    phone_book.sort()

    for i in range(0, len(phone_book)):

        # 문자열이 있는지 확인해야 한다.

        for j in range(i, len(phone_book)):

            if phone_book[j].find(phone_book[i]) >= 0:

                answer = False

                break

    return answer

두 번째 문제 풀이(정답)

문제를 잘 이해를 했으면 좋았을 것 같습니다.

접두어라는 것을 확인하고 비교하려는 문자열을 slice하여 확인하면 좀 더 시간 공간 절약이 되었을 것입니다.

def solution(phone_book):

    # 정렬을 먼저한다.

    phone_book.sort()

    print(phone_book)

    for idx in range(len(phone_book)-1):

        # 문자열이 있는지 확인한다 접두어 이기 때문에 문자열을 다 볼 필요는 없다.

        if phone_book[idx] in phone_book[idx+1][:len(phone_book[idx])]:

            return False

    return True

다른 사람 풀이 

동일한 자료형인 경우 zip으로 묶어 p1, p2를 풀 수 있습니다.

접두어를 확인하는 좋은 방법은 startswith()를 사용하는 방법이 있습니다.

def solution(phone_book):

    phone_book.sort()

    for p1, p2 in zip(phone_book, phone_book[1:]):

        if p2.startswith(p1):

            return False

    return True