취미로 코딩하기

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

이번에 볼 문제는 백준 2188번 문제인 축사 배정이다.
문제는 아래 링크를 확인하자.

www.acmicpc.net/problem/2188

문제에서 주어진 상황은 "소들의 집합"과 "축사의 집합" 두 종류의 노드에 대하여 소들끼리와 축사끼리는 연결된 간선이 없고 오직 소들과 축사 사이에만 연결된 간선(소가 선호하는 축사)이 있는 이분그래프로 나타낼 수 있다.

가장 많은 (소와 축사의 쌍)을 중복 없이 뽑아야 하는 문제이므로, 이 그래프의 maximum bipartite matching(최대 이분매칭)을 계산하는 것으로 이 문제를 해결할 수 있다.

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

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

vector<int> G[201];
int visited[201];
int matching[201];

int bimatching(int current) {
	if (visited[current]) return 0;
	visited[current] = 1;
	for (auto node : G[current]) {
		if (matching[node] == 0) {
			matching[node] = current;
			return 1;
		}
		else if (bimatching(matching[node])) {
			matching[node] = current;
			return 1;
		}
	}
	return 0;
}

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

	int N, M; cin >> N >> M;
	for (int i = 1; i <= N; i++) {
		int S; cin >> S;
		while (S--) {
			int x; cin >> x;
			G[i].push_back(x);
		}
	}

	int ans = 0;
	for (int i = 1; i <= N; i++) {
		memset(visited, 0, sizeof(visited));
		ans += bimatching(i);
	}

	cout << ans;
}