본문 바로가기
프로그래머스/Lv.2

[Python] 프로그래머스 전화번호 목록

by MINU.SHINNNN 2024. 7. 21.

https://school.programmers.co.kr/learn/courses/30/lessons/42577?language=python3

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

풀이

zip과 startswith 을 사용하면 쉽게 해결할 수 있습니다.

zip(list1, list2)는 list1, list2의 인접한 요소를 짝지어 튜플을 반환합니다. 이를 통해 반복문 내에서 인접한 두 요소를 비교할 수 있습니다. 

ex)

zip(phoneBook, phoneBook[1:])
  • phoneBook: ["123", "1234", "456"]
  • phoneBook[1:]: ["1234", "456"]

zip을 사용하면 다음과 같은 튜플들이 생성됩니다.

[("123", "1234"), ("1234", "456")]
 
그 후, string.startswith(string2) 를 사용해 접두사 검사를 진행합니다.
 
Keys:
list.sort(), zip(list1, list2), string.startswith(string)
 
def solution(phone_book):
    answer = True
    phone_book.sort()
    # Sol. 1
#     for idx, value in enumerate(phone_book):
#         if idx == len(phone_book)-1:
#             continue
#         is_find = phone_book[idx+1].find(value)
#         if is_find != -1 and is_find == 0:
#             return False
        
#     return True

    # Sol. 2 
    for p1, p2 in zip(phone_book, phone_book[1:]):
        if p2.startswith(p1):
            return False
    
    return True