[백준] java 1764 듣보잡

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

문제

김진영이 듣도 못한 사람의 명단과, 보도 못한 사람의 명단이 주어질 때, 듣도 보도 못한 사람의 명단을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M이 주어진다. 이어서 둘째 줄부터 N개의 줄에 걸쳐 듣도 못한 사람의 이름과, N+2째 줄부터 보도 못한 사람의 이름이 순서대로 주어진다. 이름은 띄어쓰기 없이 알파벳 소문자로만 이루어지며, 그 길이는 20 이하이다. N, M은 500,000 이하의 자연수이다.

듣도 못한 사람의 명단에는 중복되는 이름이 없으며, 보도 못한 사람의 명단도 마찬가지이다.

풀이 과정

문제의 출력 조건은 듣보잡을 골라 출력하는 것이다. 듣도 보도 못한 사람을 찾으려면 듣도 못한 사람 명단과 보도 못한 사람의 명단, 둘 다 존재해야할 것이다.

import java.io.*;
import java.util.*;

public class Main {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringBuilder sb = new StringBuilder();
		StringTokenizer st = new StringTokenizer(br.readLine(), " ");

		int N = Integer.parseInt(st.nextToken()) , M = Integer.parseInt(st.nextToken());
		// 명단
		Set< String > names = new HashSet<>();
		// 듣보잡 명단
		TreeSet< String > noHearAndSeeNames = new TreeSet<>();
		// 듣도 못한 사람 names 저장
		for( int i = 0; i < N; i++ )
			names.add(br.readLine());

		// 보도 못한 사람 저장
		for( int i = 0; i < M; i++ ){

			String noSeePerson = br.readLine();
			// 만약 보도못한 사람이 중복 된다면 듣보잡 명단에 추가
			if( !names.add(noSeePerson) )
				noHearAndSeeNames.add(noSeePerson);

		}

		Iterator< String > iterator = noHearAndSeeNames.iterator();
		sb.append(noHearAndSeeNames.size()).append("\n");

		while(iterator.hasNext() ){
			sb.append(iterator.next()).append("\n");
		}

		System.out.println(sb);

	}
}

우선 듣도 못한 사람 명단을 HashSet 타입인 names 변수에 저장한다. 이 후, 보도 못한 사람의 명단을 듣도 못한 사람이 저장되어 있는 names에서 대조하여, names에 있다면 듣보잡을 저장하기 위한 noHearAndSeeNames에 저장하고 출력한다.

TreeSet에 저장하는 이유는 출력 조건에 사전 순으로 출력해야한다는 조건이 있기 때문에 항상 정렬을 유지해주는 TreeSet을 사용하였다.