【圖論網絡流】【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
*/