백준 5052[자바] java 전화번호 목록

 

문제 링크: https://www.acmicpc.net/problem/5052

5052번: 전화번호 목록

 

▶문제

 

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

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

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

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

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

▶입력

 

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

▶출력

 

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

▶해설

 

자료구조 map을 이용해서 쉽게 풀 수 있는 문제입니다. 조건은 하나입니다.  

 

1. 한 번호가 다른 번호의 접두어가 되면 안 된다.

 

map을 사용하는 이유는 문자열을 하나씩 비교한다면 시간 초과가 발생합니다. 따라서 문자열들을 정렬한 후 모든 문자열을 map에 넣어줍니다. 

 

그렇다면 key 값이 문자열이 됩니다. 이제 가장 길이가 짧거나 사전 순이 낮은 문자열을 문자열을 제외하고, 다음 문자열의 0, 0~1, 0~2 씩 비교해가며 map에 key값이 존재하면 접두어입니다. 말로만 하면 어려우니 예를 보겠습니다.

 

ex) 112, 114, 11114, 11211

key값으로는 4개의 숫자가 들어갑니다. 

 

112를 제외한 114부터 진행합니다.

0 -> "1"

0~1 -> "11"

0~2 -> "114"로 map의 key값 중에서는 만족하는 값이 없습니다. 따라서 114에는 다른 문자열의 접두어가 존재하지 않습니다.  11114도 마찬가지로 진행합니다.

 

11211 진행

0 -> "1"

0~1 -> "11"

0~2 -> "112" 여기서 112의 key값이 발견됩니다. 따라서 접두어가 존재합니다. No를 추가해줍니다. 

for(int j=1; j<size; j++){
    for(int k=1; k<str[j].length(); k++){
        if(map.containsKey(str[j].substring(0,k))){
            flag = false;
            break;
        }
    }
    if(!flag){
        break;
    }
}
if(flag){
    sb.append("YES").append("\n");
}
else {
    sb.append("NO").append("\n");
}

 

전체 코드 

 

import java.awt.image.ImageProducer;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;
import java.lang.*;


public class Main {


    static int n;
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        n = Integer.parseInt(br.readLine());
        StringBuilder sb  =new StringBuilder();
        for(int i=0; i<n; i++){
            int size = Integer.parseInt(br.readLine());
            Map<String, Integer> map = new HashMap<>();
            String [] str = new String[size];

            for(int j=0; j<size; j++){
                str[j] = br.readLine();
                map.put(str[j],0);
            }

            Arrays.sort(str);

            boolean flag = true;

            for(int j=1; j<size; j++){
                for(int k=1; k<str[j].length(); k++){
                    if(map.containsKey(str[j].substring(0,k))){
                        flag = false;
                        break;
                    }
                }
                if(!flag){
                    break;
                }
            }
            if(flag){
                sb.append("YES").append("\n");
            }
            else {
                sb.append("NO").append("\n");
            }

        }

        System.out.println(sb.toString());

    }

}