(알고리즘 공부)전화번호 목록

전화번호부에 적힌 전화번호 중, 한 번호가 다른 번호의 접두어인 경우가 있는지 확인하려 합니다.
전화번호가 다음과 같을 경우, 구조대 전화번호는 영석이의 전화번호의 접두사입니다.

• 구조대 : 119
• 박준영 : 97 674 223
• 지영석 : 11 9552 4421

전화번호부에 적힌 전화번호를 담은 배열 phone_book 이 solution 함수의 매개변수로 주어질 때, 어떤 번호가 다른 번호의 접두어인 경우가 있으면 false를 그렇지 않으면 true를 return 하도록 solution 함수를 작성해주세요.

입출력

---

phone_book	return
["119", "97674223", "1195524421"]	false
["123","456","789"]	true
["12","123","1235","567","88"]	false

 

풀이

---

def solution(phone_book):
    answer = True
    phone_book.sort()
    for i in range(0,len(phone_book)):
        if i < len(phone_book)-1:
            if phone_book[i+1].startswith(phone_book[i]):
                answer = False
                break;
            else:
                answer = True    
    return answer

계속 2중 for in문을 사용해서 시간초과와 문제가 틀렸었다.

접두사에 있다면 정렬을 시킨다면 그번호 바로 다음뒤에 오게 되므로 i+1로 바로 다음 요소와 startswith로 비교하고

하나라도 있으면 false이므로 반복문에 break를 걸고 false를 리턴하게 된다.

다른사람 풀이

---

def solution(phoneBook):
    phoneBook = sorted(phoneBook)

    for p1, p2 in zip(phoneBook, phoneBook[1:]):
        if p2.startswith(p1):
            return False
    return True

 

zip을 사용해서 ["12","123","1235","567","88"]을 ("12","123"),(123","1235"),("1235","567"),("567","88")으로 바뀌면서

마지막 88은 쌍이 없어서 자동으로 없어진다. 그래서 p1, p2로 값을 접두사를 확인하면 리턴을 False 시킨다.