코딩테스트 고득점 kit, 해시 - 전화번호 목록
문제설명
전화번호부에 적힌 전화번호 중, 한 번호가 다른 번호의 접두어인 경우가 있는지 확인하려 합니다.
전화번호가 다음과 같을 경우, 구조대 전화번호는 영석이의 전화번호의 접두사입니다.
- 구조대 : 119
- 박준영 : 97 674 223
- 지영석 : 11 9552 4421
전화번호부에 적힌 전화번호를 담은 배열 phone_book 이 solution 함수의 매개변수로 주어질 때, 어떤 번호가 다른 번호의 접두어인 경우가 있으면 false를 그렇지 않으면 true를 return 하도록 solution 함수를 작성해주세요.
제한 사항
- phone_book의 길이는 1 이상 1,000,000 이하입니다.
- 각 전화번호의 길이는 1 이상 20 이하입니다.
- 같은 전화번호가 중복해서 들어있지 않습니다.
입출력 예
입출력 예 설명
예제 #1 앞에서 설명한 예와 같습니다.
예제 #2 한 번호가 다른 번호의 접두사인 경우가 없으므로, 답은 true입니다.
예제 #3 첫 번째 전화번호, “12”가 두 번째 전화번호 “123”의 접두사입니다. 따라서 답은 false입니다.
이 문제의 key포인트
-> phone_book의 길이는 1 이상 1,000,000 이하입니다.
입력범위가 크기 때문에 효율성을 고려하여 알고리즘을 작성해야 한다.
자료구조 및 알고리즘: 해시
풀이
첫번째 풀이 (정확성: O, 효율성: TC 두 개 시간초과)
def solution(phone_book):
phone_book.sort(key=len)
lenth = len(phone_book)
for i in range(lenth) :
# startswith
for j in range(lenth):
if j == i:
continue
if phone_book[i] in phone_book[j]:
if phone_book[j].startswith(phone_book[i]):
return False
answer = True
return answer특정 문자열로 시작하는지 확인하기 위해 startswith()를 사용하였다.
모든 문자열을 탐색하기 위해 이중 for문을 사용하였다.
※ phone_book의 길이가 최대 1,000,000이므로 이중 for문을 돌면 시간복잡도가 O(n^2)이 되므로 최악의 경우 1,000,000 X 1,000,000 = 1,000,000,000,000 번의 연산을 하므로 1초에 1억번의 연산을 한다고 생각하면 최악의 경우 10000초가 걸리는 알고리즘이다. 따라서 효율적이지 못한 알고리즘이 된다.
두번째 풀이 (정확성: O, 효율성: O)
def solution(phone_book):
phone_book.sort(key=len)
phone_book.sort()
lenth = len(phone_book)
for i in range(lenth - 1) :
if phone_book[i+1].startswith(phone_book[i]):
return False
return True어떤식으로 만들어야할 지 몰라서 질문하기에서 얻은 tip!
이 문제에서 for 문을 두 번 사용하면 dictionary 자료형을 사용해도 효율성을 통과하기 어렵습니다.
이 때, for 문을 한 번만 사용하기 위해서는 phone_number 리스트의 정렬이 필요합니다.
phonenumber 리스트를 값과 길이 두 가지 기준으로 정렬 하면, 반복문을 돌며 phonenumber[index] 가 phone_number[index+1]의 접두어 인지만 확인해주면 효율성 통과가 가능합니다.
phone_book을 문자열 길이로 먼저 정렬한 뒤 value값을 기준으로 정렬하면 for문을 한번만 돌리면서 그 다음 인덱스의 값이 현재 값으로 시작되는지만 확인하면 된다. 1, 2/ 2, 3/ 3, 4/ 이런식으로 탐색해가면 빠르게 탐색할 수 있다.