https://school.programmers.co.kr/learn/courses/30/lessons/42577
문제설명
전화번호부에 적힌 전화번호 중, 한 번호가 다른 번호의 접두어인 경우가 있는지 확인하려 합니다.
전화번호가 다음과 같을 경우, 구조대 전화번호는 영석이의 전화번호의 접두사입니다.
구조대 : 119
박준영 : 97 674 223
지영석 : 11 9552 4421
전화번호부에 적힌 전화번호를 담은 배열 phone_book 이 solution 함수의 매개변수로 주어질 때, 어떤 번호가 다른 번호의 접두어인 경우가 있으면 false를 그렇지 않으면 true를 return 하도록 solution 함수를 작성해주세요.
제한 사항
phone_book의 길이는 1 이상 1,000,000 이하입니다.
각 전화번호의 길이는 1 이상 20 이하입니다.
같은 전화번호가 중복해서 들어있지 않습니다.
입출력 예제
phone_book | return |
["119", "97674223", "1195524421"] | false |
["123","456","789"] | true |
["12","123","1235","567","88"] | false |
입출력 예 설명
입출력 예 #1
앞에서 설명한 예와 같습니다.
입출력 예 #2
한 번호가 다른 번호의 접두사인 경우가 없으므로, 답은 true입니다.
입출력 예 #3
첫 번째 전화번호, “12”가 두 번째 전화번호 “123”의 접두사입니다. 따라서 답은 false입니다.
문제 해결 방법
nums를 딕셔너리로 만들고, 파이썬 sort기본 정렬을 한다.
문자열에 특정 단어가 있는지 확인하기 위해 find함수를 사용하였음
코드 구현
def solution(phone_book):
answer = True
dict_phone = {}
phone_book.sort()
for i in range(len(phone_book)):
dict_phone.setdefault(phone_book[i])
for i in range(len(phone_book)):
if i == len(phone_book)-1:
break
if (phone_book[i+1].find(phone_book[i])) == -1:
pass
else:
if (phone_book[i+1].find(phone_book[i]))==0:
answer = False
return answer
else:
pass
return answer
시간/공간 복잡도
O(N)
최적화 및 개선
처음에 while문 하나로 검색하고자 하는 단어를 카운팅 하면서 nums에 있는 모든 단어들을 for문으로 조회하도록 코드를 짰는데, 효율성에서 시간초과가 나서 while문을 없애고 반복문 한번으로 조회할수 있게 해야했다.
그렇게 하기위해서 파이썬 sort기본 정렬을 사용하였다.
어려웠던 점 및 느낀점
문자열을 sort했을때 맨 앞에 문자일수록 가중치가 높아지고 만약 같은 가중치 선에서 우위를 가릴 수 없다면 그 다음 번호로 넘어가서 우위를 가린다. 나는 sort를 활용할 줄 몰랐다. 그리고 미리 sort를 하는게 이렇게 코드 효율성에 도움이 될줄 몰랐다.
파이썬 자료형 set에 대해서도 알게되었다.
'코딩테스트 > 문제 풀이 - Python' 카테고리의 다른 글
[프로그래머스 lv2]가장 큰 수 (1) | 2023.11.21 |
---|---|
[프로그래머스 lv2]의상 (0) | 2023.11.20 |
[프로그래머스 lv1]폰켓몬 (0) | 2023.11.10 |
[프로그래머스 lv1]완주하지 못한 선수 (1) | 2023.11.09 |
[프로그래머스 lv2]영어 끝말잇기 (2) | 2023.11.09 |