[Java] 백준 5052번 : 전화번호 목록

문제

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

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

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

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

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

입력

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

출력

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

예제 입력 1

2
3
911
97625999
91125426
5
113
12340
123440
12345
98346

예제 출력 1

NO
YES

 

소스코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class Main {

	private static int T, N;
	private static List<String> tel;
	public static void main(String[] args) throws NumberFormatException, IOException {

		BufferedReader bf=new BufferedReader(new InputStreamReader(System.in));
		T=Integer.parseInt(bf.readLine());
		for(int t=0;t<T;t++) {
			tel=new ArrayList<String>();
			N=Integer.parseInt(bf.readLine());
			
			for(int i=0;i<N;i++) tel.add(bf.readLine());
			Collections.sort(tel);
			
			boolean isNo=false;
			for(int i=0;i<N-1;i++) {
				int len1=tel.get(i).length();
				int len2=tel.get(i+1).length();
				if(len1<len2) {
					if(tel.get(i).equals(tel.get(i+1).substring(0, len1))) {
						isNo=true;
						break;
					}
				}
			}
			if(isNo) System.out.println("NO");
			else System.out.println("YES");
		}
	}

}

계속 시간 초과가 나서 애를 먹었던 문제입니다.

시간 초과를 해결하기 위해 아래와 같은 방법을 이용했습니다.

1. 입력 방식을 Scanner -> BufferedReader 로 변경

2. 접두어 비교 방식 변경

    처음에는 이중 for문을 이용하여 완전 탐색해주었습니다. 그러나 정렬된 전화번호 목록에서 앞뒤 두 번호의 접두어        여부만 조사하면 일관성 여부를 알 수 있습니다.

    2-1. for문을 하나만 사용하여 앞뒤 배열의 길이를 조사합니다.

        2-1-1. 앞<뒤 : substring을 이용하여 접두어 여부 확인

        2-1-2. 앞>=뒤 : for문을 종료