bzoj1433[ZJOI2009]假期的宿舍

Description

Input

Output

Sample Input

1

3

1 1 0

0 1 0

0 1 1

1 0 0

1 0 0

Sample Output

ˆ ˆ

HINT

对于30% 的数据满足1 ≤ n ≤ 12。

对于100% 的数据满足1 ≤ n ≤ 50,1 ≤ T ≤ 20。

网络流

每个学生拆点

原点向每个在校的连边（就是有床的）

汇点向每个需要床的连边（就是不在校的或者在校且没有回家的）

如果i、j认识，互相连边

然后dinic。没了

要注意i可以睡自己的床

#include<cstdio>
#include<cstring>
#define S 0
#define T 1001
#define inf 0x7fffffff
struct edge{
int to,next,v;
}e[5001];
int head[5001],h[5001];
int q[20001];
int n,cnt,ans;
inline int min(int a,int b){return a<b?a:b;}
inline void ins(int u,int v,int w)
{
e[++cnt].to=v;
e[cnt].next=head[u];
e[cnt].v=w;
head[u]=cnt;
}
inline void insert(int u,int v,int w)
{
ins(u,v,w);
ins(v,u,0);
}
inline int read()
{
int x=0;char ch=getchar();
while(ch<'0'||ch>'9')ch=getchar();
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x;
}
bool school[1001];
inline bool bfs()
{
memset(h,-1,sizeof(h));
int t=0,w=1;
q[1]=S;h[S]=0;
while (t<w)
{
int now=q[++t];
for (int j=head[now];j;j=e[j].next)
{
if (e[j].v&&h[e[j].to]==-1)
{
q[++w]=e[j].to;
h[e[j].to]=h[now]+1;
}
}
}
if (h[T]==-1) return 0;
return 1;
}
inline int dfs(int x,int f)
{
if (x==T||f==0) return f;
int w,used=0;
for (int i=head[x];i;i=e[i].next)
{
if (e[i].v&&h[e[i].to]==h[x]+1)
{
w=used;
w=dfs(e[i].to,min(e[i].v,f-used));
e[i].v-=w;
e[i^1].v+=w;
used+=w;
if (used==f) return f;
}
}
if (!used) h[x]==-1;
return used;
}
inline void dinic()
{while (bfs()) ans-=dfs(S,inf);}
inline void work()
{
ans=0;cnt=1;
memset(e,0,sizeof(e));
memset(head,0,sizeof(head));
n=read();
for (int i=1;i<=n;i++)
{
school[i]=read();
if (school[i])insert(i+n,T,1);
}
for (int i=1;i<=n;i++)
{
int x=read();
if (!school[i]||(school[i]&&!x))
{
insert(S,i,1);
ans++;
}
}
for (int i=1;i<=n;i++)
for (int j=1;j<=n;j++)
{
int x=read();
if (i==j||x)insert(i,j+n,1);
}
dinic();
if (ans==0) printf("^_^\n");
else printf("T_T\n");
}
int main()
{
int t=read();
while (t--)work();
}