[백준] S4 1822번 차집합 (JAVA)

문제 출처 - Baekjoon Online Judge

문제는 여기

1822번: 차집합

[문제] 

몇 개의 자연수로 이루어진 두 집합 A와 B가 있다. 집합 A에는 속하면서 집합 B에는 속하지 않는 모든 원소를 구하는 프로그램을 작성하시오.

[입력]

첫째 줄에는 집합 A의 원소의 개수 n(A)와 집합 B의 원소의 개수 n(B)가 빈 칸을 사이에 두고 주어진다. (1 ≤ n(A), n(B) ≤ 500,000)이 주어진다. 둘째 줄에는 집합 A의 원소가, 셋째 줄에는 집합 B의 원소가 빈 칸을 사이에 두고 주어진다. 하나의 집합의 원소는 2,147,483,647 이하의 자연수이며, 하나의 집합에 속하는 모든 원소의 값은 다르다.

[출력]

첫째 줄에 집합 A에는 속하면서 집합 B에는 속하지 않는 원소의 개수를 출력한다. 다음 줄에는 구체적인 원소를 빈 칸을 사이에 두고 증가하는 순서로 출력한다. 집합 A에는 속하면서 집합 B에는 속하지 않는 원소가 없다면 첫째 줄에 0만을 출력하면 된다.

 

 

---

[풀이]

1. A에 포함된 것 중 B에도 포함되어 있는 거라면 제거한다.

2. 포함이 된 것을 제거하는데는 Set을 사용하면 편하고 빠르게 처리할 수 있다.

3. 출력에 있어 정렬을 해야 하므로 TreeSet을 사용하는 것으로 문제를 쉽게 해결할 수 있다.

[접근]

1. 문제를 보고 Set을 사용해야 겠다는 생각을 하였다.

2. 결과를 출력하는데 있어 Size만 출력하는 것이 아닌 원소들도 오름차순으로 출력해야 하므로 TreeSet을 사용했다.

[코드]

package BOJ_silver;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.TreeSet;
import java.util.StringTokenizer;

public class Main_S4_1822 {
	static int a; // a의 개수
	static int b; // b의 개수
	static TreeSet<Integer> set;

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		
		a = Integer.parseInt(st.nextToken());
		b = Integer.parseInt(st.nextToken());
		
		set = new TreeSet<>();
		
		// A에 포함이 되어있다면 추가
		st = new StringTokenizer(br.readLine());
		for (int i = 0; i < a; i++) {
			set.add(Integer.parseInt(st.nextToken()));
		}
		
		// B에 포함이 되어있다면 제거
		st = new StringTokenizer(br.readLine());
		for (int i = 0; i < b; i++) {
			int n = Integer.parseInt(st.nextToken());
			if (set.contains(n))
				set.remove(n);
		}
		
		// 차집합의 크기
		System.out.println(set.size());
		// 차집합의 내용
		for(int n : set)
            System.out.print(n + " ");
	}
}