【BZOJ1433】【codevs2347】假期的宿舍，最大流

传送门1

传送门2

写在前面：蛤蛤

思路：

（要拆点啊！naive的我一开始没拆点竟然还过了4个点）

每个人拆成两个点，一个在源点那里，一个在汇点那里

1.s连本校学生，容量为1的边（发床了）

2.没回家的本校学生和外面的人往t连容量为1的边（需要床睡觉）

3.认识的人（包括自己的点）之间连容量为1的边

跑出最大流，判断其是否与需要的床数相等即可

注意：拆点啊！拆点

代码：

#include<bits/stdc++.h>
using namespace std;
int n,tot,s,t,T;
int first[125],dis[125];
bool a[65];
struct edge
{
int v,w,next;
}e[100000];
void add(int x,int y,int z){e[++tot]=(edge){y,z,first[x]};first[x]=tot;}
queue<int>q;
bool bfs()
{
memset(dis,0,sizeof(dis));
q.push(s);dis[s]=1;
while (!q.empty())
{
int k=q.front();
q.pop();
for (int i=first[k];i;i=e[i].next)
if(!dis[e[i].v]&&e[i].w)
q.push(e[i].v),dis[e[i].v]=dis[k]+1;
}
return dis[t];
}
int dfs(int x,int maxn)
{
if (x==t) return maxn;
int used=0;
for (int i=first[x];i;i=e[i].next)
if(dis[e[i].v]==dis[x]+1)
{
int k=dfs(e[i].v,min(maxn-used,e[i].w));
e[i].w-=k;e[i^1].w+=k;
used+=k;
if (used==maxn) return maxn;
}
if (!used) dis[x]=0;
return used;
}
main()
{
int x;
scanf("%d",&T);
while (T--)
{
memset(first,0,sizeof(first));
scanf("%d",&n);
int ans=n;
t=2*n+1;tot=1;
for (int i=1;i<=n;i++)
{
scanf("%d",a+i);
if (a[i])
add(s,i,1),
add(i,s,0);
else add(i+n,t,1),add(t,i+n,0);
}
for (int i=1;i<=n;i++)
{
scanf("%d",&x);
if (a[i])
if (!x) add(i+n,t,1),add(t,i+n,0);
else ans--;
}
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
{
scanf("%d",&x);
if (x||i==j) add(i,j+n,1),add(j+n,i,0);
}
while (bfs()) ans-=dfs(s,0x7fffff);
if (ans) puts("T_T");
else puts("^_^");
}
}