[Baekjoon/C++] 1269번 - 대칭 차집합

[백준] Baekjoon Online Judge

문제로 이동

 

문제

자연수를 원소로 갖는 공집합이 아닌 두 집합 A와 B가 있다. 이때, 두 집합의 대칭 차집합의 원소의 개수를 출력하는 프로그램을 작성하시오. 두 집합 A와 B가 있을 때, (A-B)와 (B-A)의 합집합을 A와 B의 대칭 차집합이라고 한다.

예를 들어, A = { 1, 2, 4 } 이고, B = { 2, 3, 4, 5, 6 } 라고 할 때,  A-B = { 1 } 이고, B-A = { 3, 5, 6 } 이므로, 대칭 차집합의 원소의 개수는 1 + 3 = 4개이다.

 

입력

첫째 줄에 집합 A의 원소의 개수와 집합 B의 원소의 개수가 빈 칸을 사이에 두고 주어진다. 둘째 줄에는 집합 A의 모든 원소가, 셋째 줄에는 집합 B의 모든 원소가 빈 칸을 사이에 두고 각각 주어진다. 각 집합의 원소의 개수는 200,000을 넘지 않으며, 모든 원소의 값은 100,000,000을 넘지 않는다.

 

출력

첫째 줄에 대칭 차집합의 원소의 개수를 출력한다.

 

---

예제 입력	예제 출력
3 5 1 2 4 2 3 4 5 6	4

---

풀이

#include <iostream>
#include <unordered_map>
using namespace std;

unordered_map <int, bool> m;

int main() {
	int A, B, n;
	scanf("%d %d", &A, &B);

	for (int i = 0; i < A; i++) {
		scanf("%d", &n);

		m.insert({ n, 0 });
	}

	int same = 0;

	for (int i = 0; i < B; i++) {
		scanf("%d", &n);

		// 해당 원소가 이미 존재하는지 체크
		if (m.count(n) != 0) same++;
	}

	printf("%d\n", A + B - (same * 2));

	return 0;
}

집합 A의 원소들은 map에 넣은 뒤 입력 받은 집합 B의 원소들이 A에 포함되어 있는지 확인하는 방식으로 코드를 짰다. 집합 B의 원소들은 map에 굳이 넣을 필요가 없기에 map에 insert하지 않았다.