[JAVA] 백준 1822번- 차집합

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

1822번: 차집합

 

import java.util.*;
public class Main{
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int a = sc.nextInt();
        int b = sc.nextInt();
        int[] arrA = new int[a];
        int[] arrB = new int[b];

        TreeSet<Integer> aSet = new TreeSet<>();

        for(int i=0; i<a; i++){
            aSet.add(sc.nextInt());
        }
        for(int i=0; i<b; i++){
            int k = sc.nextInt();
            if(aSet.contains(k)) aSet.remove(k);
        }
        System.out.println(aSet.size());
        for(int x:aSet){
            System.out.print(x+" ");
        }

    }
}

 

a집합을 입력받아 모두 TreeSet에 추가한다. 

그 다음 b집합 원소들을 입력받아 TreeSet에 이미 있다면 제거한다.