취미로 코딩하기

※ 글쓴이는 취미로 코딩을 익혀보는 사람이라 정확하지 않은 내용을 담고 있을 수 있다 ※

이번에 볼 문제는 백준 1043번 문제인 거짓말이다.
문제는 아래 링크를 확인하자.

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

이 문제에서는 지민이가 진실을 말할 수 있는 파티의 수를 세야한다.

Disjoint Set 자료구조를 이용하면, (지민이를 제외한) 파티에서 만난 사람들끼리 만나고 다시 만나는 모임을 하나의 집합으로 간단하게 모을 수 있다. 그러면, 각 파티의 모임의 대표와 진실을 아는 사람을 한 사람씩 저장해두고 모든 파티를 disjoint set에 넣어 계산한 뒤 진실을 아는 사람과 연결되지 않은 파티의 수를 마지막으로 따로 세어 출력하는 것으로 이 문제를 해결할 수 있다.

아래는 제출한 소스코드이다.

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

int arr[51];
int query[51];
int findf(int x) {
	if (arr[x] == x) return x;
	else return arr[x] = findf(arr[x]);
}

void unionf(int x, int y) {
	x = findf(x), y = findf(y);
	if (x == y) return;
	arr[y] = x;
}

int main() {
	ios::sync_with_stdio(0);
	cin.tie(0);

	int N, M; cin >> N >> M;
	for (int i = 1; i <= N; i++) arr[i] = i;
	int T; cin >> T;
	if (T == 0) {
		cout << M;
		return 0;
	}

	T--;
	int truth; cin >> truth;
	while (T--) {
		int x; cin >> x;
		unionf(truth, x);
	}

	int MM = M;
	while (MM--) {
		int K, x; cin >> K >> x;
		K--; query[MM] = x;
		while (K--) {
			int y; cin >> y;
			unionf(x, y);
		}
	}

	int ans = 0;
	for (int i = 0; i < M; i++) {
		if (findf(query[i]) != findf(truth)) ans++;
	}

	cout << ans;
}