bzoj 1433: [ZJOI2009]假期的宿舍 -- 最大流

1433: [ZJOI2009]假期的宿舍

Time Limit: 10 Sec  Memory Limit: 162 MB

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。

我们从源点向所有住宿的人连边，从所有的床位连向汇点，然后再从人连向他所认识的人的床位

这样建边，然后跑最大流

这样答案如果和住宿人数相等则满足，否则不满足

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
#define inf 100007
#define F(i) for(i=1;i<=n;i++)
#define N 50010
#define M 110
inline int rd()
{
    int x=,f=;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
    return x*f;
}
int lj[M],fro[N],to[N],cnt,v[N],S,T=;
inline void add(int a,int b,int c){fro[++cnt]=lj[a];to[cnt]=b;v[cnt]=c;lj[a]=cnt;}
inline void ins(int a,int b,int c){add(a,b,c);add(b,a,);}
int dis[M],q[N],h,t;
bool bfs()
{
    memset(dis,,sizeof(dis));
    dis[]=h=t=;q[]=;
    int tp;
    while(h<=t)
    {
        tp=q[h++];
        for(int i=lj[tp];i;i=fro[i])
        {
            if(!dis[to[i]]&&v[i])
            {
                dis[to[i]]=dis[tp]+;
                q[++t]=to[i];
            }
        }
    }
    return dis[T]?:;
}
int dfs(int x,int p)
{
    if(x==T) return p;
    int tp,res=;
    for(int i=lj[x];i;i=fro[i])
    {
        if(v[i]&&dis[to[i]]==dis[x]+)
        {
            tp=dfs(to[i],min(p-res,v[i]));
            v[i]-=tp;
            v[i^]+=tp;
            res+=tp;
            if(res==p) return p;
        }
    }
    if(res==)  dis[x]=;
    return res;
}
int n,tt,sc[M],tot,ans;
int main()
{
    tt=rd();
    int i,j,x;
    while(tt--)
    {
        tot=ans=;
        memset(lj,,sizeof(lj));cnt=;
        n=rd();
        F(i){
            sc[i]=rd();
            if(sc[i]) ins(i+n,T,);
        }
        F(i){
            x=rd();
            if((sc[i]&&!x)||!sc[i])
            {
                ins(,i,);
                tot++;
            }
        }
        F(i) F(j)
        {
            x=rd();
            if(x||i==j) ins(i,j+n,);
        }
        while(bfs()) ans+=dfs(,inf);
        if(tot==ans) puts("^_^");
        else puts("T_T");
    }
}