[BOJ/Silver 3] 백준 1269 대칭 차집합(C++)

문제 링크

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

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

알고리즘 분류

• 자료 구조

풀이

set 자료구조를 이용하여 A의 원소가 B 집합에 있는 지 찾고, B의 원소가 A 집합에 있는 지 찾아서 없다면 카운트를 1 증가시킨다. A, B 집합을 탐색하면서 증가시킨 카운트가 답이다.

코드

#include <iostream>
#include <string>
#include <cmath>
#include <unordered_set>
#include <unordered_map>
#include <bitset>
#include <ctime>
#include <stack>
#include <queue>
#include <vector>
#include <algorithm>
#include <climits>
#define FASTIO cin.tie(NULL); cout.tie(NULL); ios::sync_with_stdio(false);
#define MAX 100001
#define LL long long
#define INF 1e9

using namespace std;
int A, B;
unordered_set<int> SetA, SetB;
unordered_set<int>::iterator IT;
int Answer = 0;

void Input() {
	cin >> A >> B;
	while (A--) {
		int N;
		cin >> N;
		SetA.insert(N);
	};
	while (B--) {
		int N;
		cin >> N;
		SetB.insert(N);
	};
}

void Settings() {
	for (auto IT = SetA.begin(); IT != SetA.end(); IT++) {
		int Cur = *IT;
		if (SetB.find(Cur) == SetB.end()) {
			Answer++;
		}
	}
	for (auto IT = SetB.begin(); IT != SetB.end(); IT++) {
		int Cur = *IT;
		if (SetA.find(Cur) == SetA.end()) {
			Answer++;
		}
	}
}

void Find_Answer() {
	cout << Answer << "\n";
}

int main() {
	FASTIO
	Input();
	Settings();
	Find_Answer();

	return 0;
}