프로그래머스 - 전화번호 목록 (Hash) / Level 2

문제

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

• 구조대 : 119
• 박준영 : 97 674 223
• 지영석 : 11 9552 4421

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

 

제한 사항

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

 

입출력 예제

phone_number	return
["119", "97674223", "1195524421"]	false
["123","456","789"]	true
["12","123","1235","567","88"]	false

 

입출력 예 #1
앞에서 설명한 예와 같습니다.

 

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

 

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

 

 

풀이

def solution(phone_book):
    answer = True
    phone_book.sort()  # 문자열들을 문자순으로 정렬. 앞에서 부터 같은 문자열을 포함하는 것들은 서로 인접할 수 밖에 없다.

    for i in range(len(phone_book)):

        if i != len(phone_book) - 1:
            target = phone_book[i + 1][:len(phone_book[i])]  # target은 다음 문자열을 현재 문자열 크기로 자른 것.

            if target == phone_book[i]:  # 인접한 두 문자열을 같은 크기로 비교.
                answer = False
                return answer

    return answer

 

가장 간단하게 생각한다면 문자열들을 알맞게 슬라이싱 해서 전부 비교해보면 된다.

이는 이중 for문으로 구성되어 O(n^2)의 시간 복잡도를 갖게 되는데

역시나 해당 풀이법은 프로그래머스의 효율성 검사에서 통과하지 못했다.

이유는 n의 크기가 최대 1,000,000 이기 때문이다.

 

그렇다면 시간복잡도를 n^2 미만으로 떨어뜨리기 위해 여러 생각을 해봤다.

그중에서 테스트 케이스를 통과한 위 풀이법은 sort 함수의 특성을 이용하여

단일 for문으로 알고리즘을 구성한 것이다.

 

여러 문자열들이 들어있는 list를 sort 함수를 이용하여 정렬하게 되면 사전 순으로 정렬되게 된다.

예를 들어, ["1234", "12", "121334", "1209845"]를  sort 하게 되면

다음과 같이 된다. ["12", "1209845", "121334", "1234"]

 

따라서 특정 문자열과 해당 문자열을 접두사로 포함하는 문자열은 인접해있다.

그러므로 단일 for문 내에서 인접한 요소들끼리만 비교하면 된다.

 

 

ps. 해당 문제는 Hash 카테고리에 속하는 문제인데 Hash를 이용한 더 좋은 풀이법이 있으면

공유해주시면 감사하겠습니다!