[programmers] 전화번호 목록 - Java

전화번호부에 적힌 전화번호 중 한 번호가 다른 번호의 접두어인 경우를 찾는 문제를 풀기 위한 두 가지 코드가 제시되었습니다. 첫 번째 코드는 두 개의 반복문을 사용하여 모든 전화번호를 비교하는 방식으로, 시간 복잡도가 O(n^2)로 큰 전화번호부에서는 실행 시간이 길어집니다. 개선된 코드는 전화번호부를 사전순으로 정렬한 후, 각 전화번호가 바로 다음 번호의 접두어인지만 확인하는 방식으로, 시간 복잡도가 O(n log n)으로 훨씬 효율적입니다.

DriedPollack

Mar 28, 2024

Contents

전화번호 목록 처음 시도한 코드 개선한 코드 핵심 키워드 결론!

전화번호 목록

문제 설명

전화번호부에 적힌 전화번호 중, 한 번호가 다른 번호의 접두어인 경우가 있는지 확인하려 합니다.

전화번호가 다음과 같을 경우, 구조대 전화번호는 영석이의 전화번호의 접두사입니다.

구조대 : 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

입출력 예 설명

입출력 예 #1

앞에서 설명한 예와 같습니다.

입출력 예 #2

한 번호가 다른 번호의 접두사인 경우가 없으므로, 답은 true입니다.

입출력 예 #3

첫 번째 전화번호, “12”가 두 번째 전화번호 “123”의 접두사입니다. 따라서 답은 false입니다.

처음 시도한 코드


class Solution {
    public boolean solution(String[] phone_book) {
        boolean answer = true;
        for(int i=0; i<phone_book.length; i++){
            String str = phone_book[i];
            for(int j=0; j<phone_book.length; j++){
                if(j==i || str.length()<phone_book[j].length())
                    continue;
                else{
                    if(str.startsWith(phone_book[j]))
                        return false;
                }
                  
            }
        }
        return answer;
    }
}

개선한 코드


import java.util.Arrays;

class Solution {
    public boolean solution(String[] phone_book) {
        Arrays.sort(phone_book); // 전화번호부를 사전순으로 정렬
        
        for (int i = 0; i < phone_book.length - 1; i++) {
            if (phone_book[i + 1].startsWith(phone_book[i])) { // 다음 번호가 현재 번호의 접두어인지 확인
                return false;
            }
        }
        
        return true;
    }
}

핵심 키워드

처음 작성한 코드는 두 개의 반복문을 사용하여 각 전화번호를 모든 다른 전화번호와 비교했다.

이 경우 O(n^2)의 시간 복잡도를 가지며, 전화번호부의 크기가 크면 실행 시간이 길어졌다.

이 문제를 해결하기 위해 개선한 코드는 전화번호부를 사전순으로 정렬한 후, 각 전화번호가 바로 다음 번호의 접두어인지만 확인했다.

수정된 코드는 전체적인 시간 복잡도가 O(n log n)으로 훨씬 효율적이다. 따라서 효율성 테스트를 통과할 수 있었다.

결론!

해당 문제를 풀면서 이중 반복문을 사용하지 않고도 같은 결과를 낼 수 있는 방법을 생각해 볼 수 있었다.

☕Java 💡Coding Test