[프로그래머스] 전화번호 목록(Java/Python)

문제설명

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

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

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

제한사항

• phone_book의 길이는 1 이상 1,000,000 이하입니다.
  • 각 전화번호의 길이는 1 이상 20 이하입니다.
  • 같은 전화번호가 중복해서 들어있지 않습니다.

전체코드(Java)

import java.util.*;
class Solution {
    public boolean solution(String[] phone_book) {
        boolean answer = true;
        // 전화번호부를 사전순으로 정렬
        Arrays.sort(phone_book);
        // 전화번호부 하나 앞과 현재를 비교하며, 접두사 일경우 false를 return
        // 자신의 앞과 현재를 비교하는건 "123","1234"일 경우 자신의 접두사가 앞에 있을때를 고려한 경우
        for(int loop=1;loop<phone_book.length;loop++){
            String shortStr = phone_book[loop-1].length() > phone_book[loop].length() ? phone_book[loop] : phone_book[loop-1];
            String longStr = phone_book[loop-1].length() > phone_book[loop].length() ? phone_book[loop-1] : phone_book[loop];
            if(longStr.indexOf(shortStr)==0){
                return false;
            }
        }
        return answer;
    }
}

전체코드(Python)

def solution(phone_book):
    answer = True
    phone_book = sorted(phone_book)
    length = len(phone_book)
    # 전화번호부를 사전순으로 정렬
    for i in range(1,length):
        shortStr = phone_book[i-1] if len(phone_book[i]) > len(phone_book[i-1]) else phone_book[i]
        longStr = phone_book[i] if len(phone_book[i]) > len(phone_book[i-1]) else phone_book[i-1]
        # 전화번호부 하나 앞과 현재를 비교하며, 접두사 일경우 False를 return
        # 자신의 앞과 현재를 비교하는건 "123","1234"일 경우 자신의 접두사가 앞에 있을때를 고려한 경우
        if longStr.find(shortStr) == 0:
            return False
    return answer

 

혹시라도 정정할 내용이나 추가적으로 필요하신 정보가 있다면 댓글 남겨주시면 감사하겠습니다.

오늘도 Jindory 블로그에 방문해주셔서 감사합니다.