[백준] 1822번 - 차집합 [Java]

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

 

[백준] 1822번 - 차집합 [Java]

[백준] 1822번 - 차집합 [Java]

---

1.  아이디어

 

집합과 정렬을 동시에 수행할 수 있는 TreeSet을 활용했다.

---

2. 문제풀이

 

집합 A를 TreeSet으로 선언 후 집합 A의 원소들을 A에 넣은 후 집합 B에 해당하는 원소들을 A에서 제거하면 차집합을 얻을 수 있는 점을 이용해서 구현했다.

---

3. 코드

 

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));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        StringBuilder sb = new StringBuilder();
        StringTokenizer st = new StringTokenizer(br.readLine());

        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        Set<Integer> A = new TreeSet<>();
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            A.add(Integer.parseInt(st.nextToken()));
        }

        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < M; i++) {
            A.remove(Integer.parseInt(st.nextToken()));
        }

        sb.append(A.size()).append("\n");
        for (int n : A) {
            sb.append(n).append(" ");
        }

        bw.write(sb.toString());
        bw.flush();
    }
}

---

4. 후기