BZOJ 1433 ZJOI2009 假期的宿舍 最大流

题目大意：给定一些人，有些人是在校学生，有些去学校探访，在校学生有些回家，一个人只能睡认识的人的床，求能不能睡下

二分图的模型，左侧是所有需要睡觉的人，右侧是所有能用的床铺，二分图最大匹配即可

嫌建图麻烦可以考虑最大流

一个点拆成两个

如果这个人需要睡床，从原点出发向这个人的第一个点连边

如果这个人是在校学生，从这个人的第二个点向汇点连边

如果i==j或者i和j认识，从i的第一个点出发向j的第二个点连边

跑最大流判断是否满流即可

#include 
#include 
#include 
#include 
#define M 110
#define INF 0x3f3f3f3f
#define S 0
#define T (n+n+1)
using namespace std;
struct abcd{
	int to,f,next;
}table[100100];
int head[M],tot=1;
int n,ans,at_school[M],go_home[M];
void Initialize()
{
	memset(head,0,sizeof head);
	tot=1;ans=0;
}
void Add(int x,int y,int z)
{
	table[++tot].to=y;
	table[tot].f=z;
	table[tot].next=head[x];
	head[x]=tot;
}
void Link(int x,int y,int z)
{
	Add(x,y,z);
	Add(y,x,0);
}
namespace Max_Flow{
	int dpt[M];
	bool BFS()
	{
		static int q[M];
		int i,r=0,h=0;
		memset(dpt,-1,sizeof dpt);
		q[++r]=S;dpt[S]=1;
		while(r!=h)
		{
			int x=q[++h];
			for(i=head[x];i;i=table[i].next)
				if(table[i].f&&!~dpt[table[i].to])
				{
					dpt[table[i].to]=dpt[x]+1;
					q[++r]=table[i].to;
					if(table[i].to==T)
						return true;
				}
		}
		return false;
	}
	int Dinic(int x,int flow)
	{
		int i,left=flow;
		if(x==T) return flow;
		for(i=head[x];i&&left;i=table[i].next)
			if(table[i].f&&dpt[table[i].to]==dpt[x]+1)
			{
				int temp=Dinic(table[i].to,min(left,table[i].f) );
				if(!temp) dpt[table[i].to]=-1;
				left-=temp;
				table[i].f-=temp;
				table[i^1].f+=temp;
			}
		return flow-left;
	}
}
int main()
{
	int cases,i,j,x;
	for(cin>>cases;cases;cases--)
	{
		Initialize();
		cin>>n;
		for(i=1;i<=n;i++)
		{
			scanf("%d",&at_school[i]);
			if(at_school[i]) Link(n+i,T,1);
		}
		for(i=1;i<=n;i++)
		{
			scanf("%d",&go_home[i]);
			if(!at_school[i]||!go_home[i])
				Link(S,i,1),++ans;
		}
		for(i=1;i<=n;i++)
			for(j=1;j<=n;j++)
			{
				scanf("%d",&x);
				if(x||i==j) Link(i,j+n,1);
			}
		while( Max_Flow::BFS() )
			ans-=Max_Flow::Dinic(S,INF);
		if(!ans)puts("^_^");
		else puts("T_T");
	}
}