[Baekjoon] 1764 듣보잡 - JAVA

문제

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

---

입력

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

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

---

출력

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

---

예제 입력 1 

3 4
ohhenrie
charlie
baesangwook
obama
baesangwook
ohhenrie
clinton

예제 출력 1 

2
baesangwook
ohhenrie

 

---

📝 제출 답안

import java.io.*;
import java.util.HashSet;
import java.util.StringTokenizer;
import java.util.TreeSet;

public class Main {
    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());

        HashSet<String> notHeard = new HashSet<>();
        for (int i = 0; i < n; i++) {
            String input = br.readLine();
            notHeard.add(input);
        }

        TreeSet<String> res = new TreeSet<>();
        for (int i = 0; i < m; i++) {
            String input = br.readLine();
            if (notHeard.contains(input)) {
                res.add(input);
            }
        }

        StringBuilder sb = new StringBuilder();
        sb.append(res.size()).append("\n");
        for (String s : res) {
            sb.append(s).append("\n");
        }

        System.out.println(sb);
    }
}

 

이번 문제는 두 개의 집합을 생성했다.

 

먼저 기본 HashSet으로 n명의 듣도 못한 사람 명단을 저장하고, m명의 보도 못한 사람 이름을 입력 받음과 동시에

듣도 못한 사람 명단에 있으면 출력하고자 하는 TreeSet으로 선언한 res 집합에 저장한다.

 

HashSet은 빠른 검색에 최적화되어 있어 평균 시간복잡도가 O(1)이다.

 

TreeSet이 자동으로 오름차순으로 정렬하여 저장되는 집합이라 사전순으로 출력하기 위해 사용해주었다.

 

TreeSet을 순회하면서 StringBuilder에 문자열을 효율적으로 추가하고 출력해준다.

 

제출 현황