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

https://www.acmicpc.net/problem/5052

5052번: 전화번호 목록

풀이

테스트 케이스 수를 입력받습니다.

입력받을 전화번호 목록의 갯수를 입력받고 String 배열을 선언해줍니다.

String 배열에 전화번호 목록을 입력받습니다.

 

Arrays.sort()를 하여 배열을 정렬을 해 준 후

길이가 가장 짧은거부터 다음 문자열의 처음 부분에 해당되면 일관성이 없어 다른사람에게 연락이 갈 수 있기 때문에 "NO"를 출력해주고, 아닐경우 "YES"를 출력해주면 됩니다.

 

소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.Comparator;

public class Main {
    public static void main(String[] args) throws IOException{
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in)); 
    
        int size = Integer.parseInt(br.readLine());
        
        StringBuilder sb = new StringBuilder();
        while(size --> 0) {
            int num = Integer.parseInt(br.readLine());
            
            String[] strs = new String[num];
            
            for(int i = 0; i < num; i++) {
                strs[i] = br.readLine();
            }
            
            Arrays.sort(strs, new Comparator<String>() {
                public int compare(String o1, String o2) {
                    return o1.compareTo(o2);
                }
            });
            
            String answer = "YES";
            for(int i = 1; i < num; i++) {
                if(strs[i].startsWith(strs[i-1])) {
                    answer = "NO";
                    break;
                }
            }
            sb.append(answer).append("\n");
        }
        System.out.print(sb);
    }
}