题目大意:给定一些人,有些人是在校学生,有些去学校探访,在校学生有些回家,一个人只能睡认识的人的床,求能不能睡下
二分图的模型,左侧是所有需要睡觉的人,右侧是所有能用的床铺,二分图最大匹配即可
嫌建图麻烦可以考虑最大流
一个点拆成两个
如果这个人需要睡床,从原点出发向这个人的第一个点连边
如果这个人是在校学生,从这个人的第二个点向汇点连边
如果i==j或者i和j认识,从i的第一个点出发向j的第二个点连边
跑最大流判断是否满流即可
#include <cstdio> #include <cstring> #include <iostream> #include <algorithm> #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"); } }