HDU 二分图专题

今天学习了下二分图匹配以及判断是否为二分图的算法。二分图匹配的思路是，先匹配着，不行就让之前的人想办法挪位置，挪不了了就结束；判断是否为二分图的思路是使用BFS实现交叉染色，将有边相连的两个节点染成不同的颜色。下面分别看一下这两个部分。

一、二分图匹配

1. HDU 2063 过山车

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2063
代码如下，具体见注释：

#include <iostream>
#include <string.h>
using namespace std;
//HDU	Accepted	2063	15MS	2404K	729B	G++
int m, n;  //m: 女生的人数，n: 男生的人数
int cp[505][505];  //意向cp关系
int girl[505];  //存的是给第i号女生分配的男生编号
bool used[505];  //是否已分配
int find(int x) {
	for(int i=1; i<=m; i++) {  //i从1到女生个数m进行遍历
		if(cp[x][i] == 1 && used[i] == 0) {  //x对i号有意愿并且i号女生还未分配出去
			used[i] = 1;
			if(girl[i] == 0 || find(girl[i])) {  //女生未被选择过或被该女生选择的男生还可以选择找其他女生
				girl[i] = x;  //给第i号女生分配第x号男生
				return 1;
			}
		}
	}
	return 0;
}

int main() {
	int k, a, b;
	while(cin>>k && k) {
		cin>>m>>n;
		memset(girl, 0, sizeof(girl));
		memset(cp, 0, sizeof(cp));
		for(int i=1; i<=k; i++) {
			cin>>a>>b;
			cp[b][a] = 1;  //存关系
		}
		int sum = 0;  //答案
		for(int i=1; i<=n; i++) {
			memset(used, 0, sizeof(used));  //used数组表示：在匹配编号为i的男生时，女生访问情况，每次尝试匹配新的一个男生是都要重新置零
			if(find(i) == 1)
				sum++;
		}
		cout<<sum<<endl;
	}
	
	return 0;
}

2. HDU 1083 Courses

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=1083
题意是有p个课程和n个学生，每个课程都对应一些学生，问能否选出学生代表，使得所有的课程都有一个学生作为代表。这是一个二分图匹配的问题。若最大匹配等于p，则说明可以，输出"YES"；否则输出"NO"。
AC代码，详见注释：

#include <iostream>
#include <string.h>
using namespace std;
const int MAXN = 400;
//HDU	Accepted	1083	1123MS	2032K	792 B	G++
//二分图匹配
int t, p, n;
int mp[MAXN][MAXN];  //保存输入的相互关系
bool mark[MAXN];  //是否查找过
int rec[MAXN];  //保存匹配关系

bool find(int x) {  //对这个（x号）课程找匹配的学生 
	int i;
	for(i=1; i<=n; i++) {  //遍历学生 
		if(!mark[i] && mp[x][i] == 1) {  //这个mark数组在每次查找x之前都要清空，但是注意不能在find函数中清空，否则会死循环 
			mark[i] = 1;  //标记为已访问 
			if(rec[i] == -1 || find(rec[i])) {  //这个学生还没有匹配给其他课程，或者这个学生匹配的课程还可以匹配到其他学生 
				rec[i] = x;
				return true;
			}
		}
	}
	return false;
}

int main() {
	cin>>t;
	while(t--) {
		memset(mp, 0, sizeof(mp));
		memset(rec, -1, sizeof(rec));
		cin>>p>>n;
		int j, k;
		for(int i=1; i<=p; i++) {
			cin>>k;
			while(k--) {
				cin>>j;
				mp[i][j] = 1;
			}
		}
		int cnt = 0;
		for(int i=1; i<=p; i++) {
			memset(mark, 0, sizeof(mark));
			if(find(i)) {
				cnt++;
			}
		}
		if(cnt == p)
			cout<<"YES"<<endl;
		else
			cout<<"NO"<<endl;
	}
	return 0;
} 

二、判断是否为二分图

1. HDU 4751 Divide Groups

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4751
题意：学校举行校友会，校友之间的关系可能是互相认识或单向认识。问能否将将校友分成两组，使得每组校友都互相认识。
解题思路：使用交叉染色法，首先将校友关系使用邻接矩阵mp[i][j]记录，再使用一个vis[i]数组记录染色情况。vis数组初始值全为-1，代表还未被染色。当要给u这一节点及其不认识的节点染色时，首先将vis[u]染为0值，转到BFS函数中对能染色的节点进行染色。BFS中的染色规则为如果两人不认识，且另一人i未被染色，那么i被染成与u相反的颜色。（异或运算，1^1 = 0, 0^1 = 1）。如果在染色过程中发现节点i被染色，但是颜色与u是相同的，则说明不是二分图。
AC代码：

#include <iostream>
#include <string.h>
#include <queue>
using namespace std;
const int MAXN = 110;
//HDU	Accepted	4751	78MS	1412K	988 B	G++
int n;
bool mp[MAXN][MAXN];
int vis[MAXN];

int bfs(int x) {
	queue<int> q;
	while(!q.empty()) 	q.pop();
	q.push(x);
	while(!q.empty()) {
		int u = q.front();
		q.pop();
		for(int i=1; i<=n; i++) {
			if(i == u || (mp[u][i] && mp[i][u]))	continue;
			if(vis[i] == -1) {
				//未着色
				vis[i] = vis[u] ^ 1;  //取与u相反的颜色
				q.push(i);
			}
			else if(vis[u] == vis[i])
				return false;
		}
	}
	return true;
}

int main() {
	while(cin>>n && n) {
		int j;
		memset(mp, 0, sizeof(mp));
		memset(vis, -1, sizeof(vis));
		for(int i=1; i<=n; i++) {
			while(cin>>j && j) {
				mp[i][j] = 1;
			}
		}
		bool flag = false;
		for(int i=1; i<=n; i++) {
			if(vis[i] == -1) { //未染色 
				vis[i] = 0;
				if(!bfs(i)) {
					flag = 1;
					break;
				}
			}
		}
		if(flag)	cout<<"NO"<<endl;
		else	cout<<"YES"<<endl;
	} 
	return 0;
}

2. HDU 2444 The Accomodation of Students

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2444
题意：将N个人分成两组，同组间任意两人互不认识，如果不能分成两组，则输出No，否则输出两组间两个人互相认识的对数。
解题思路：首选判断是否是二分图，若是，再求最大匹配数。
AC代码：

#include <iostream>
#include <algorithm>
#include <string.h>
#include <cmath>
#include <queue>
using namespace std;
//需要复习 
//HDU	Accepted	2444	78MS	5316K	1366 B	G++ 
typedef long long int LL;
const int maxn = 1000;
int g[maxn][maxn], used[maxn], link[maxn];
int color[maxn];
int nx, ny, n, m;

//二分匹配 
bool dfs(int u) {
	for(int v = 1; v <= n; v++) {
		if(g[u][v] && !used[v]) {
			used[v] = 1;
			if(link[v] == -1 || dfs(link[v])) {
				link[v] = u;
				return true;
			}
		}
	}
	return false;
}

int maxmatch() {
	int res = 0;
	memset(link, -1, sizeof(link));
	for(int i=1; i<=n; i++) {
		memset(used, 0, sizeof(used));
		if(dfs(i))	res++;
	}
	return res / 2;  //最后答案是我们算的最大匹配数的一半 
}

//染色
bool dye(int u) {
	for(int v=1; v<=n; v++) {
		if(g[u][v]) {
			if(color[v] == -1) {
				color[v] = !color[u];
				if(!dye(v))
					return false;
			}
			else if(color[v] == color[u])
				return false;
		}
	}
	return true;
}

int main() {
	while(~scanf("%d%d", &n, &m)) {
		memset(g, 0, sizeof(g));
		for (int i=0; i<m; i++) {
			int u, v;
			scanf("%d%d", &u, &v);
			g[u][v] = g[v][u] = 1;  //建图
		}
		
		memset (color, -1, sizeof (color));
		int ok = 1;
		for (int i=1; i<=n; i++) {  //染色，可能不止一个连通分量
			if(color[i] == -1) {
				color[i] = 1;
				if(!dye(i)) {
					ok = 0;
					break;
				}
			}
		}
		if (ok) printf ("%d\n", maxmatch ());
		else printf ("No\n");
	}
	return 0;
}