[백준 알고리즘] 백준 1822번 차집합 자바(Java)

@notepad_jj2

츄르사려고 코딩하는 코집사입니다.

---

1. [백준 알고리즘] 백준 1822번 차집합 자바(Java)

1) 문제번호 : 1822번

 

2) 문제 출처

www.acmicpc.net/problem/1822

1822번: 차집합

 

2. 문제

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

 

3. 제약사항

- 

4. 입력

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

 

5. 출력

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

 

6. 풀이

- 집합의 요소가 중복되지 않게 해주고, 자동으로 정렬해주는 TreeSet이나 HashSet을 사용한다.

- 집합 B의 원소를 입력하면서 만약에 aList 안에 입력받는 B의 원소가 있으면 그 원소를 삭제하기.

7. 소스 코드

import java.util.*;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		int aNum = sc.nextInt(); //집합 A 원소의 개수
		int bNum = sc.nextInt(); //집합 B 원소의 개수
		
		//집합 A 선언
		TreeSet<Integer> aList = new TreeSet<Integer>();
		
		//집합 A 원소 입력
		for(int i=0;i<aNum;i++) aList.add(sc.nextInt());
		
		//집합 B 원소 입력
		for(int i=0;i<bNum;i++) {
			int num = sc.nextInt();
			if(aList.contains(num)) aList.remove(num);
		}
		
		System.out.println(aList.size());
		for(Integer c : aList) System.out.print(c + " ");
	}
}

---