[BackJoon]1764번, TreeSet<T>

문제

 

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

 

입력

 

첫째 줄에 듣도 못한 사람의 수 N, 보도 못한 사람의 수 M이 주어진다. 

이어서 둘째 줄부터 N개의 줄에 걸쳐 듣도 못한 사람의 이름과, N+2째 줄부터 보도 못한 사람의 이름이 순서대로 주어진다. 

이름은 띄어쓰기 없이 알파벳 소문자로만 이루어지며, 그 길이는 20 이하이다. 

N, M은 500,000 이하의 자연수이다.

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

 

출력

 

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

 

풀이

 

처음엔 일단 받아서, 답을 구한 후 정렬해서 스트링빌더에 넣어 출력하는 방향을 생각했으나

 

애초에 자료구조 차원에서 정렬이 이루어진다면 그럴 필요가 없다는 생각이 들어 그냥 출력해 버렸다.

 

그래도 문제풀이 속도를 위해 기준으로 삼을 N 집합만 트리 셋으로 만들고 M 집합은 해시 셋으로 만들었다.

 

이후에 주어진 개수만큼 원소를 채운 뒤 N 집합을 기준으로 교집합 연산을 진행,

 

이어 트리셋에 의해 정렬이 끝난 교집합의 크기와 원소를 그대로 출력해 주었다.

package BackJoon;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.Set;
import java.util.StringTokenizer;
import java.util.TreeSet;

public class Prob1764_2 {

    public static void main(String[] args) throws IOException {

        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        StringTokenizer st = new StringTokenizer(br.readLine());

        int n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());

        Set<String> setN = new TreeSet<>();
        Set<String> setM = new HashSet<>();

        for (int i = 0; i < n; i++) {
            setN.add(br.readLine());
        }

        for (int i = 0; i < m; i++) {
            setM.add(br.readLine());
        }

        setN.retainAll(setM);

        System.out.println(setN.size());

        for (String result : setN) {
            System.out.println(result);
        }
    }
}