[BOJ] 백준 5052번 : 전화번호 목록 (JAVA)

문제

전화번호 목록이 주어진다. 이때, 이 목록이 일관성이 있는지 없는지를 구하는 프로그램을 작성하시오.

전화번호 목록이 일관성을 유지하려면, 한 번호가 다른 번호의 접두어인 경우가 없어야 한다.

예를 들어, 전화번호 목록이 아래와 같은 경우를 생각해보자

• 긴급전화: 911
• 상근: 97 625 999
• 선영: 91 12 54 26

이 경우에 선영이에게 전화를 걸 수 있는 방법이 없다. 전화기를 들고 선영이 번호의 처음 세 자리를 누르는 순간 바로 긴급전화가 걸리기 때문이다. 따라서, 이 목록은 일관성이 없는 목록이다.

 

 

입력

첫째 줄에 테스트 케이스의 개수 t가 주어진다. (1 ≤ t ≤ 50) 각 테스트 케이스의 첫째 줄에는 전화번호의 수 n이 주어진다. (1 ≤ n ≤ 10000) 다음 n개의 줄에는 목록에 포함되어 있는 전화번호가 하나씩 주어진다. 전화번호의 길이는 길어야 10자리이며, 목록에 있는 두 전화번호가 같은 경우는 없다.

 

 

출력

각 테스트 케이스에 대해서, 일관성 있는 목록인 경우에는 YES, 아닌 경우에는 NO를 출력한다.

 

 

풀이

프로그래머스의 전화 번호 목록 문제와 상당히 유사한 문제였습니다. 프로그래머스 버전 전화 번호 목록에 대한 풀이가 궁금하신 분은 이 곳을 눌러주시면 되겠습니다.

 

먼저, 알고리즘 분류 상 해싱, 트라이, 정렬 이렇게 있었지만, 사실 정렬과 java String 클래스의 startsWith 메소드만 알고있어도 쉽게 풀 수 있는 문제였습니다.

 

여기서, startsWith는 대상 문자열이 특정 문자 또는 문자열로 시작하는지 체크하는 함수이며, "12345".startsWith("123")을 입력하면 true가 반환됩니다.

 

다만, 무턱대고 O(N^2)으로 모든 요소에 대해서 문자열이 서로 접두어 관계인지 확인하면 시간 초과가 나기때문에 좀 더 똑똑한 방식을 사용해야 했습니다.

 

간단합니다. 바로 정렬을 이용하는 것이죠. 전화번호 목록이 오름차순 정렬되어있고, 일관성이 있지 않다고 가정해 봅시다.

 

일관성이 없다는 것은 특정 문자열 2개 이상이 접두어 관계를 가진다는 것을 의미합니다. 이 때, 접두어 관계를 가지는 문자열들은 필연적으로 서로 붙어있습니다. ex) ["123", "1244", "125", "37"]

 

따라서, 우리가 해야할 일은 단순해집니다. 바로, 특정 문자열과 그 다음 문자열에 startsWith 메소드를 이용하여 접두어 관계를 가지는지 확인하는 것이죠.

 

아래는 위 과정을 정리한 소스코드입니다.

 

소스코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.Arrays;
 
public class Main {
 
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        int T = Integer.parseInt(br.readLine());
 
        while (T-- > 0) {
            int N = Integer.parseInt(br.readLine());
            String[] phone_number = new String[N]; // 전화번호 목록.
 
            for (int i = 0; i < N; i++) {
                phone_number[i] = br.readLine();
            }
            Arrays.sort(phone_number); // 오름차순으로 정렬.
 
            if (isConsistent(N, phone_number)) {
                bw.write("YES\n");
            } else {
                bw.write("NO\n");
            }
        }
 
        bw.flush();
        bw.close();
        br.close();
    }
 
    // 전화번호 목록이 일관성이 있는지 검사.
    public static boolean isConsistent(int N, String[] phone_number) {
        // 현재 배열은 오름차순 정렬이 되어있는 상태.
        // 만약 전화 번호 목록 안에 접두어 관계가 있는 문자열이 있다면,
        // 특정 문자열 바로 뒤에 접두어 관계가 있는 문자열이 있을 것이다.
        // 결과적으로, 전화번호 목록이 일관성이 있는지 검사하려면
        // 특정 문자열과 그 다음 문자열의 접두어 관계만 확인하면 된다.
        for (int i = 0; i < N - 1; i++) {
            if (phone_number[i + 1].startsWith(phone_number[i])) {
                return false;
            }
        }
 
        return true;
    }
 
}

 

지적 혹은 조언 환영합니다! 언제든지 댓글로 남겨주세요.