BZOJ 1433 [ZJOI2009]假期的宿舍

默默地断更好久，总算还是决定厚着脸皮先来水一篇续上。。。

orz Chester_king帮我发现初始化问题。。。

#include<cstdio>
#include<vector>
#include<queue>
#include<cstring>
#define INF 1e8
using namespace std;

struct Edge{
int from,to,cap,flow;
};

vector <Edge> edges;
vector <int> G[105];

int T,s,t,n,m,a[51],b[51],d[105],v[105],cur[105];

void add(int x,int y,int cap,int flow)
{
edges.push_back((Edge){x,y,cap,0});
G[x].push_back(m);
++m;
edges.push_back((Edge){y,x,0,0});
G[y].push_back(m);
++m;
}

int BFS(int s,int t)
{
memset(d,0,sizeof(d));
memset(v,0,sizeof(v));
d[s]=0;
v[s]=1;
queue<int>dui;
dui.push(s);
while(!dui.empty())
{
int p=dui.front();
dui.pop();
for(int i=0;i<G[p].size();++i)
{
Edge e=edges[G[p][i]];
if(!v[e.to]&&e.cap>e.flow)
{
v[e.to]=1;
d[e.to]=d[p]+1;
dui.push(e.to);
}
}
}
return v[t];
}

int min(int a,int b){return a<b?a:b;}

int DFS(int x,int a)
{
if(x==t||a==0)
return a;
int f,flow=0;
for(int &i=cur[x];i<G[x].size();++i)
{
Edge &e=edges[G[x][i]];
if(d[e.to]==d[x]+1&&(f=DFS(e.to,min(a,e.cap-e.flow))>0))
{
e.flow+=f;
a-=f;
edges[G[x][i]^1].flow-=f;
flow+=f;
if(a==0)
break;
}
}
return flow;
}

int maxflow(int s,int t)
{
int flow=0;
while(BFS(s,t))
{
memset(cur,0,sizeof(cur));
flow+=DFS(s,INF);
}
return flow;
}

int main()
{
scanf("%d",&T);
while(T--)
{
edges.clear();
scanf("%d",&n);
int i,j,p,tmp=0;
for(i=0;i<=104;++i)
G[i].clear();
m=0;
s=0,t=104;
for(i=1;i<=n;++i)
{
scanf("%d",&a[i]);
if(a[i])
add(i+n,t,1,0); //这个人有床，连一条边
}
for(i=1;i<=n;++i)
{
scanf("%d",&b[i]);
if((a[i]&&!b[i])||!a[i]) //这个人不是在校生或者是回家的在校生就连一条边，他要住在学校里
{
++tmp;
add(s,i,1,0);
}
}
for(i=1;i<=n;++i)
for(j=1;j<=n;++j)
{
scanf("%d",&p);
if((p&&a[j])||i==j) //i和j认识并且j有床或者是他自己连一条边
add(i,j+n,1,0);
}
if(maxflow(s,t)==tmp)
printf("^_^\n");
else
printf("T_T\n");
}
return 0;
}

写给自己：考试爆炸心态很崩？不行那就不是我了。赶紧开始搞事，做上进的好学生。毕竟爆炸了代表进步空间很大（逃）