[백준 C++] 1269. 대칭 차집합

문제링크

문제

자연수를 원소로 갖는 공집합이 아닌 두 집합 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을 넘지 않는다.

출력

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

예제 입력 / 출력

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

// 예제 출력 1
4

풀이

📍 알고리즘

• 차집합의 개수는 각 원소가 상대 집합에 속하냐 안속하냐로 구할 수 있다.

Ex. A = { 1, 2, 4 }, B = { 2, 3, 4, 5, 6 }

❗ 여기서 A - B의 개수를 구하는 방법은 B집합에 포함된 A 원소의 개수를 A의 총 원소의 개수에서 빼주면 된다.

➡ 2, 4는 B 집합에서 찾아 볼 수 있기 때문에 A의 총 원소 개수인 3에서 2를 빼주면 A - B의 개수인 1을 구할 수 있음
➡ B - A도 동일한 방식으로 구할 수 있음

#include <iostream>
#include <unordered_set>

using namespace std;


int main(void) {

	unordered_set<int> A, B;
	int A_size, B_size, temp;

	cin >> A_size >> B_size;

	for (int i = 0; i < A_size; i++) {
		cin >> temp;
		A.insert(temp);
	}
	
	for (int i = 0; i < B_size; i++) {
		cin >> temp;
		B.insert(temp);
	}

	for (auto i : A) {
		if (B.find(i) != B.end()) A_size--;
	}

	for (auto i : B) {
		if (A.find(i) != A.end()) B_size--;
	}

	cout << A_size + B_size;
}