洛谷2055 假期的宿舍(网络流)
最近氵谷怎么一直推网络流。。。
【题目分析】
良心的数据范围,肯定就是O()的玩意儿了,emmm,网络流!
因为题目中给出了许多关系,最后询问是否能全部解决,那么就相当于给了匹配关系,询问最大匹配数与总数之间的关系,所以网络流的解法就出来了。
对于每个非在校学生和要回家的学生,肯定就不需要考虑他们了,将他们与s连容量为1的边限制最多1个人,对于每个住校学生,肯定需要床,那么就将自己的床与t连容量为1的边,因为自己可以睡自己的床,那么和自己的床连边。再读入人和人的关系,如果为1就将这个人和那个人的床连容量为1的边。
最后跑一遍最大流询问最大匹配数,如果最大匹配数小于住校学生数,那么就不可以,否则可以。
PS:记得初始化。。。。。。
【代码~】
#include<bits/stdc++.h>
using namespace std;
const int MAXN=201;
const int MAXM=1e4+10;
const int INF=0x3f3f3f3f;
int n,m,cnt,s,t,total;
int head[MAXN],cur[MAXN],depth[MAXN];
int nxt[MAXM],to[MAXM],w[MAXM];
int stu[MAXN],hom[MAXN];
int Read()
{
int i=0,f=1;
char c;
for(c=getchar();(c>'9'||c<'0')&&c!='-';c=getchar());
if(c=='-')
f=-1,c=getchar();
for(;c>='0'&&c<='9';c=getchar())
i=(i<<3)+(i<<1)+c-'0';
return i*f;
}
void Add(int x,int y,int z)
{
nxt[cnt]=head[x];
head[x]=cnt;
to[cnt]=y;
w[cnt]=z;
cnt++;
}
void add(int x,int y,int z)
{
Add(x,y,z);
Add(y,x,0);
}
bool bfs()
{
memset(depth,0,sizeof(depth));
queue<int> q;
depth[s]=1;
q.push(s);
while(!q.empty())
{
int u=q.front();
q.pop();
for(int i=head[u];i!=-1;i=nxt[i])
{
int v=to[i];
if(!depth[v]&&w[i])
{
depth[v]=depth[u]+1;
q.push(v);
}
}
}
return depth[t]!=0;
}
int dfs(int u,int flow)
{
if(u==t)
return flow;
for(int &i=cur[u];i!=-1;i=nxt[i])
{
int v=to[i];
if(depth[v]==depth[u]+1&&w[i])
{
int di=dfs(v,min(w[i],flow));
if(di>0)
{
w[i]-=di;
w[i^1]+=di;
return di;
}
}
}
return 0;
}
int dinic()
{
int ans=0;
while(bfs())
{
for(int i=0;i<=t;++i)
cur[i]=head[i];
while(int d=dfs(s,INF))
ans+=d;
}
return ans;
}
int main()
{
int T=Read();
while(T--)
{
cnt=0,total=0;
memset(head,-1,sizeof(head));
memset(stu,0,sizeof(stu));
memset(hom,0,sizeof(hom));
n=Read();
s=0,t=n+n+1;
for(int i=1;i<=n;++i)
stu[i]=Read();
for(int i=1;i<=n;++i)
hom[i]=Read();
for(int i=1;i<=n;++i)
{
if(!stu[i]||(stu[i]&&!hom[i]))
add(s,i,1),total++;
if(stu[i])
add(i+n,t,1);
}
for(int i=1;i<=n;++i)
{
if(stu[i]&&!hom[i])
add(i,i+n,1);
for(int j=1;j<=n;++j)
{
int x=Read();
if(x)
add(i,j+n,1);
}
}
if(dinic()==total)
puts("^_^");
else
puts("T_T");
}
return 0;
}