【BZOJ 1433】 [ZJOI2009]假期的宿舍

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
#define MAXN 510
#define INF 100000000
bool b[MAXN];
int g[MAXN+2][MAXN+2];
int n,T,s,t;
int team[MAXN+10],head,tail;
int d[MAXN];
bool bfs()
{
head=tail=0;memset(d,0,sizeof(d));
d[s]=1;team[++tail]=s;
while(head<tail)
{
int x=team[++head];
for(int i=1;i<=n;i++)
if(d[i]==0&&g[x][i])
d[i]=d[x]+1,team[++tail]=i;
}
if(d[t]==0) return false;
return true;
}
int dfs(int x,int mmin)
{
if(x==t) return mmin;
int tmp,f=0;
for(int i=1;i<=n;i++)
if(d[i]==d[x]+1&&g[x][i]&&(tmp=dfs(i,min(mmin,g[x][i]))))
{
g[x][i]-=tmp;
g[i][x]+=tmp;
mmin-=tmp,f+=tmp;
if(mmin==0) return f;
}
return f;
}
int main()
{
scanf("%d",&T);
while(T--)
{
memset(g,0,sizeof(g));
memset(b,false,sizeof(b));
scanf("%d",&n);
s=n*2+1,t=n*2+2;
int tmp,sum=0,ans=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&tmp);
if(tmp==0) g[s][i]=1,sum++;
if(tmp==1) g[i][i+n]=1,g[i+n][t]=1,b[i]=true;
}
for(int i=1;i<=n;i++)
{
scanf("%d",&tmp);
if(b[i]&&tmp==0) g[s][i]=1,sum++;
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
scanf("%d",&tmp);
if(tmp==1) g[i][j+n]=1;
}
n*=2;
n+=2;
while(bfs()) ans+=dfs(s,INF);
if(ans==sum) printf("%s\n","^_^");
else         printf("%s\n","T_T");
}
return 0;
}

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
using namespace std;
#define MAXN 510
#define INF 100000000
bool b[MAXN];
int n,T,s,t;
int team[MAXN+10],head,tail;
int d[MAXN];
int tot=2,g[25000],num[25000],nnext[25000],flow[25000];
void Add(int x,int y,int z){nnext[tot]=g[x];g[x]=tot;num[tot]=y;flow[tot]=z;tot++;}
bool bfs()
{
head=tail=0;memset(d,0,sizeof(d));
d[s]=1;team[++tail]=s;
while(head<tail)
{
int x=team[++head];
for(int i=g[x];i;i=nnext[i])
{
int tmp=num[i];
if(d[tmp]==0&&flow[i]!=0)
d[tmp]=d[x]+1,team[++tail]=tmp;
}
}
if(d[t]==0) return false;
return true;
}
int dfs(int x,int mmin)
{
if(x==t) return mmin;
int tmp,f=0;
for(int i=g[x];i;i=nnext[i])
if(d[num[i]]==d[x]+1&&flow[i]!=0&&(tmp=dfs(num[i],min(mmin,flow[i]))))
{
flow[i]-=tmp;
flow[i^1]+=tmp;
mmin-=tmp,f+=tmp;
if(mmin==0) return f;
return tmp;
}
return f;
}
int main()
{
scanf("%d",&T);
while(T--)
{
memset(b,false,sizeof(b));
memset(g,0,sizeof(g));
tot=2;
scanf("%d",&n);
s=n*2+1,t=n*2+2;
int tmp,sum=0,ans=0;
for(int i=1;i<=n;i++)
{
scanf("%d",&tmp);
if(tmp==0)
{
Add(s,i,1);
Add(i,s,0);
sum++;
}
if(tmp==1)
{
b[i]=true;
Add(i,i+n,1);
Add(i+n,i,0);
Add(i+n,t,1);
Add(t,i+n,0);
}
}
for(int i=1;i<=n;i++)
{
scanf("%d",&tmp);
if(b[i]==true&&tmp==0)
{
Add(s,i,1);
Add(i,s,0);
sum++;
}
}
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
scanf("%d",&tmp);
if(tmp==1)
{
Add(i,j+n,1);
Add(j+n,i,0);
}
}
while(bfs()) ans+=dfs(s,INF);
if(ans==sum) printf("%s\n","^_^");
else         printf("%s\n","T_T");
}
return 0;
}