백준 1269 대칭 차집합 (자료구조 실버 4)

문제

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

 

출력

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

 

내 제출

#include <stdio.h>
#include <stdlib.h>

int BinarySearch(int* A, int size, int find) {
	int first = 0, last = size - 1;

	while (first <= last) {
		int mid = (first + last) / 2;

		if (A[mid] == find) {
			return 1;
		}
		else if (A[mid] < find) {
			first = mid + 1;
		}
		else if (A[mid] > find) {
			last = mid - 1;
		}
	}
	return 0;
}

int compare(const void* a, const void* b) {
	return *((int*)a) - *((int*)b);
}

int main(void) {
	int N, M;
    scanf("%d %d", &N, &M);

	int* A = malloc(sizeof(int) * N);
	for (int i = 0; i < N; i++) {
		scanf("%d", &A[i]);
	}
	qsort(A, N, sizeof(int), compare);

	int count = 0;
	for (int i = 0; i < M; i++) {
		int find; scanf("%d", &find);
		count += BinarySearch(A, N, find);
	}

	printf("%d\n", N + M - count*2);

	return 0;
}

교집합 원소를 찾고, 여기에 2를 곱한 뒤 총 원소의 개수에서 빼준 값을 출력하면 된다.

집합 A, B의 원소의 개수를 N, M으로 입력받는다.
그 후 A 배열을 동적 할당 해준 다음 원소를 입력받고 qsort를 이용해 정렬을 한다.
M개의 입력(집합 B의 원소)이 들어오는 동안 이진 탐색을 진행한다.

두 원소를 더한 뒤 교집합 원소에 2를 곱해서 빼면 정답이다.