찬우의 이것저것 Chanwoo's blog

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

2020-04-16

https://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

string.startswith("aa") # 이 함수를 사용하면 aa로 시작하는 문자열을 찾아준다.

해당 함수를 사용할줄 알면 굉장히 쉽게 풀 수 있다.

def solution(phone_book):
    phone_book = sorted(phone_book)
    for p1, p2 in zip(phone_book[:-1],phone_book[1:]):
        if p2.startswith(p1):
            return False
    return True

다른사람의 풀이를 보다가 정규표현식을 사용해서 푼걸 보았다.

^는 문자열의 시작부터 일치하는게 있는지 찾는것이기에 startswith("str")과 동일한 기능을 한다.

import re
def solution(phoneBook):

    for b in phoneBook:
        p = re.compile("^"+b)
        for b2 in phoneBook:
            if b != b2 and p.match(b2):
                return False
    return True

Related Posts