https://school.programmers.co.kr/learn/courses/30/lessons/42577
프로그래머스
코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.
programmers.co.kr
문제 설명
전화번호부에 적힌 전화번호 중, 한 번호가 다른 번호의 접두어인 경우가 있는지 확인하려 합니다.
전화번호가 다음과 같을 경우, 구조대 전화번호는 영석이의 전화번호의 접두사입니다.
- 구조대 : 119
- 박준영 : 97 674 223
- 지영석 : 11 9552 4421
전화번호부에 적힌 전화번호를 담은 배열 phone_book 이 solution 함수의 매개변수로 주어질 때, 어떤 번호가 다른 번호의 접두어인 경우가 있으면 false를 그렇지 않으면 true를 return 하도록 solution 함수를 작성해주세요.
제한 사항
- phone_book의 길이는 1 이상 1,000,000 이하입니다.
풀이 과정
방법 1: sort 이용
def solution(phone_book):
phone_book.sort()
for i in range(len(phone_book)-1):
if phone_book[i] == phone_book[i+1][:len(phone_book[i])]:
return False
return True
- 만약 전화번호 a가 전화번호 b의 접두사라면, phone_book을 오름차순으로 sort했을 때, a는 b의 왼쪽에 위치하게 된다.
- 따라서 phone_book을 sort한 후, 첫 번째 원소부터 차례로 자신의 오른쪽 원소와 비교하여, 오른쪽 원소의 접두사가 자기 자신과 같은 경우 false를 return하도록 한다.
방법2: sort와 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
- 방법1과 같은 아이디어인데, phone_book[i] == phone_book[i+1][:len(phone_book[i])] 대신 zip과 startswith을 활용한다.
방법 3: hash 이용
def solution(phone_book):
# 1. Hash map 생성
hash_map = {}
for phone_number in phone_book:
hash_map[phone_number] = 1
# 2. 접두어가 Hash map에 존재하는지 찾기
for phone_number in phone_book:
prefix = ""
for number in phone_number:
prefix += number
# 3. 자기 자신 제외
if prefix in hash_map and prefix != phone_number:
return False
return True
- phone_number 내 각각의 번호를 key로 하고 value가 1인 hash map을 생성한 후, 각각의 번호의 접두어가 hash map에 존재하는지 찾는다.
'Algorithm > 프로그래머스' 카테고리의 다른 글
[프로그래머스] 해시 - 위장 (0) | 2023.04.06 |
---|---|
[프로그래머스] 탐욕법(Greedy) - 체육복 (0) | 2023.03.15 |
댓글