HDU3081 Marriage Match II【网络流】

题意：n个女生，n个男生，每一轮女生选一个男生（他们之间没有争吵过且前几轮没有选过的），女生还可以选她朋友能选的男生（朋友的朋友是朋友），组成完备匹配，问最多进行几轮

思路：最大流，源点 →女生 → 男生 → 汇点，二分轮数k，源点出去的边容量为k，到汇点的也为k，女生、男生建好所有可能的边，容量为1，最大流为k*n的话，能进行k轮匹配。用并查集记录女生之间的关系

#include<stdio.h>
#include<iostream>
#include<string.h>
#include<string>
#include<stdlib.h>
#include<math.h>
#include<vector>
#include<list>
#include<map>
#include<set>
#include<stack>
#include<queue>
#include<algorithm>
#include<numeric>
#include<functional>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int maxn = 300;
struct edge {int to,cf,rev;};
vector<edge> G[maxn];
int lev[maxn],iter[maxn];
int pre[maxn],dis[maxn][maxn];

void init(int x)
{
	memset(dis,0,sizeof dis);
	for(int i = 0; i <= x; i++)
		pre[i] = i;	
}

void init2(int x)
{
	for(int i = 0; i <= x; i++)
		G[i].clear();
}

int fid(int x)
{
	if(pre[x] == x)
		return x;
	else
		return pre[x] = fid(pre[x]);
}

void he(int x,int y)
{
	int nx = fid(x);
	int ny = fid(y);
	if(nx != ny)
		pre[nx] = ny;
}

void add(int from, int to, int cap)
{
	G[from].push_back((edge){to,cap,G[to].size()});
	G[to].push_back((edge){from,0,G[from].size()-1});
}

void bfs(int s)
{
	memset(lev,-1,sizeof lev);
	queue<int> q;
	lev[s] = 0;
	q.push(s);
	while(!q.empty())
	{
		int v = q.front();q.pop();
		for(int i = 0; i < G[v].size(); i++)
		{
			edge &e = G[v][i];
			if(e.cf > 0 && lev[e.to] < 0)
			{
				lev[e.to] = lev[v] + 1;
				q.push(e.to);
			}
		}
	}
}

int dfs(int v,int t, int f)
{
	if(v == t) return f;
	for(int &i = iter[v]; i < G[v].size(); i++)
	{
		edge &e = G[v][i];
		if(e.cf > 0 && lev[v] < lev[e.to])
		{
			int d = dfs(e.to, t, min(f,e.cf));
			if(d > 0)
			{
				e.cf -= d;
				G[e.to][e.rev].cf += d;
				return d;
			}
		}
	}
	return 0;
}

int maxflow(int s,int t)
{
	int flow = 0;
	while(1)
	{
		bfs(s);
		if(lev[t] < 0) return flow;
		memset(iter,0,sizeof iter);
		int f;
		while((f = dfs(s,t,0x3f3f3f3f)) > 0)
			flow += f;
	}
}

int check(int k,int n)
{
	init2(2*n+1);
	for(int i = 1; i <= n; i++)
	{
		for(int j = 1; j <= n; j++)
		{
			if(dis[i][j]) add(i,j+n,1);
		}
	}
	for(int i = 1; i <= n; i++)
		add(0,i,k);
	for(int i = 1+n; i <= 2*n; i++)
		add(i,2*n+1,k);
	int ans = maxflow(0,2*n+1);
	if(ans == n*k)
		return 1;
	else
		return 0;
}

int main(void)
{
	int T,n,m,f;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d%d",&n,&m,&f);
		init(n);
		while(m--)
		{
			int a,b;
			scanf("%d%d",&a,&b);
			dis[a][b] = 1;
		}
		while(f--)
		{
			int a,b;
			scanf("%d%d",&a,&b);
			he(a,b);
		}
		for(int i = 1; i <= n; i++)
		{
			for(int j = i+1; j <= n; j++)
			{
				if(fid(i) == fid(j))
				{
					for(int k = 1; k <= n; k++)
						dis[i][k] = dis[j][k] = (dis[i][k] || dis[j][k]);
				}
			}
		}
		int l = 0,r = n;
		while(l < r)
		{
			int mid = (l+r+1)/2;
			if(check(mid,n))
				l = mid;
			else
				r = mid-1;
		}
		printf("%d\n",l);
	}
	return 0;
}