[백준] 1764번: 듣보잡 Java

문제

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

 

입력

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

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

 

출력

듣보잡의 수와 그 명단을 사전순으로 출력한다.

 

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

1764번: 듣보잡

접근

두 집합의 교집합을 묻는 문제입니다.

먼저 듣도 못한 사람들의 데이터를 저장하고

보도 못한 사람이 듣도 못한 사람의 집합에 존재하면

제출할 배열 혹은 리스트에 저장합니다.

명단은 사전순으로 출력해야 함으로 정렬한 뒤 이를 출력합니다.

 

듣도 못한 사람들의 데이터를 저장할 때,

list를 사용하고 보도 못한 사람을 contian()을 통해 유무를 검사하게 된다면,

시간초과가 발생합니다.

 

데이터의 빠른 검색을 위해 HashMap이나 HashSet을 사용하여 구현합니다.

 

구현

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.*;

public class Main {
	//https://www.acmicpc.net/problem/1764
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
		StringTokenizer st=new StringTokenizer(br.readLine());
		int n=Integer.parseInt(st.nextToken()); //듣도 못한 사람 수
		int m=Integer.parseInt(st.nextToken()); //보도 못한 사람 수
		
		Map<String, Integer> map = new HashMap<>();//듣보잡 판별을 위한 맵
		List<String> list = new ArrayList<String>();// 답 제출
		
		//2번 이름이 불리면 듣보잡
		for(int i=0;i<n+m;i++) {
			String name = br.readLine();
			//한번만 불릴 경우
			if(!map.containsKey(name)) {
				map.put(name, 1);
			//두번 불릴 경우 듣보잡    
			}else {
				list.add(name);
			}
		}
		//듣보잡 수 출력
		bw.append(list.size()+"\n");
		//사전순 정렬
		Collections.sort(list);
		//듣보잡 명단 출력
		for(String s:list) {
			bw.append(s+"\n");
		}
		bw.close();
		br.close();
	}
}