【图论网络流】【CQBZOJ 2433】Dormitory

问题 I(2433): Dormitory
时间限制: 1 Sec 内存限制: 128 MB

题目描述
小蛮妹子有许多的，什么棉花糖啦，蜥蜴啊，小乌龟啊，只有你想不到的没有她没有的。小蛮妹子的小宠物都住在一个漂亮的大屋子里，每个人都有自己的床位，但是由于小蛮妹子的宠物太多了，以至于这些宠物之间并不是都互相认识。
这些宠物们也有自己的朋友，这些朋友一直羡慕他们住的漂亮大屋子，于是有一天，小蛮妹子出去玩了，一些小宠物也偷偷出去玩了，那些留守的小宠物们开始招待来访的客人。
这些来客都打算在这里住一晚上，可是安排床位就出了麻烦。每个小宠物都只愿意在自己的直接朋友或自己的床上（如果它是小蛮的宠物）睡觉，且一物睡一张床。
询问是否能有一个方案来满足所有在留宿的宠物都能住宿。

输入
第一行一个数 T 表示数据组数。
接下来 T 组数据。
每组数据第一行一个数 n 表示涉及到的总宠物数（包括小蛮的宠物和来客）。
接下来一行 n 个数，第 i 个数表示第 i 个个宠物是否是在小蛮的宠物 (0 表示不是，1 表示是)。再接下来一行 n 个数，第 i 个数表示第 i 个人是否出去玩(0 表示不出去，1 表示出去)注意如果第 i 个人不是小蛮的宠物，那么这个位置上的数是一个随机的数，你应该在读入以后忽略它)。接下来 n 行，每行 n 个数，第 i 行第 j 个数表示 i 和 j 是否认识 (1 表示认识，0 表示不认识，第 i 行 i 个的值为 0，但是显然自己还是可以睡自己的床)，认识的关系是相互的。

输出
每一行对应每组数据
若存在方案满足条件 输出 ^.^ ，否则输出 T.T （只有三个字符，没有空格等其他内容）

样例输入
1
3
1 1 0
0 1 0
0 1 1
1 0 0
1 0 0

样例输出
^.^

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

因为要求是否存在一个方案使得每一个动物都有床，我们不妨用网络流，把源点向需要床的动物连边，床向汇点连边，需要床的动物向自己的床或者朋友的床连边，若最大流跑出来刚好为需要床的动物的数量，那么可行，否自不行。

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;

#define MAXN 50
#define MAXM
#define INF 0x3f3f3f3f
typedef long long int LL;

int getint()
{
    int rn=0;
    char c=getchar();

    while(c<'0'||'9'<c)
        c=getchar();

    while('0'<=c&&c<='9')
    {
        rn=rn*10+c-'0';
        c=getchar();
    }

    return rn;
}

int N,S,E,T;
bool havebed[MAXN*2+10],leave[MAXN*2+10];
int c[MAXN*2+10][MAXN*2+10];

int d[MAXN*2+10];
int vd[MAXN*2+10];

int aug(int x,int augco)
{
    if(x==E)return augco;

    int augc=augco,delta,Dmin=T-1;

    for(int j=1;j<=T;++j)
        if(c[x][j]>0)
        {
            if(d[x]==d[j]+1)
            {
                delta=min(augc,c[x][j]);
                delta=aug(j,delta);

                c[x][j]-=delta;
                c[j][x]+=delta;
                augc-=delta;

                if(d[S]>=T)
                    return augco-augc;

                if(augc==0)
                    break;
            }

            Dmin=min(Dmin,d[j]);
        }

    if(augco==augc)
    {
        --vd[d[x]];
        if(vd[d[x]]==0)
            d[S]=T;

        d[x]=Dmin+1;
        ++vd[d[x]];
    }

    return augco-augc;
}

int isap()
{
    memset(d,0,sizeof(d));
    memset(vd,0,sizeof(vd));
    vd[0]=T;

    int flow=0;
    while(d[S]<T)
        flow+=aug(S,INF);

    return flow;
}

int main()
{
    //freopen("dormitory3.in","r",stdin);

    int Case;
    scanf("%d",&Case);

    while(Case--)
    {
        memset(havebed,0,sizeof(havebed));
        memset(leave,0,sizeof(leave));
        memset(c,0,sizeof(c));

        scanf("%d",&N);
        S=N*2+1,E=N*2+2,T=N*2+2;
        int i,j,sum=0,tmp;

        for(i=1;i<=N;++i)
        {
            havebed[i]=getint();

            if(havebed[i])
                c[N+i][E]=1;
        }

        for(i=1;i<=N;++i)
        {
            leave[i]=getint();

            if(!leave[i]||!havebed[i])
            {
                ++sum;
                c[S][i]=1;
            }
        }

        for(i=1;i<=N;++i)
            for(j=1;j<=N;++j)
                if(getint()||i==j)c[i][N+j]=1;

        puts(isap()>=sum?"^.^":"T.T");
    }
}

/*
1
3
1 1 0
0 1 0
0 1 0
1 0 1
0 1 0
*/