[BZOJ1433][ZJOI2009]假期的宿舍 二分图匹配

题目链接：http://www.lydsy.com/JudgeOnline/problem.php?id=1433

首先留在学校的学生向自己的床连边。

要住在学校里的人向认识的学生的床连边。

跑二分图匹配，看匹配的数量是否等于住在学校的人数。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int inline readint(){
    int Num;char ch;
    while((ch=getchar())<'0'||ch>'9');Num=ch-'0';
    while((ch=getchar())>='0'&&ch<='9') Num=Num*10+ch-'0';
    return Num;
}
int N;
bool A[55],B[55];
int to[2510],ne[2510],fir[110],cnt;
void Add(int a,int b){
    to[++cnt]=b;
    ne[cnt]=fir[a];
    fir[a]=cnt;
}
bool vis[110];
int match[110];
int Dfs(int u){
    for(int i=fir[u];i!=-1;i=ne[i]){
        int v=to[i];
        if(!vis[v]){
            vis[v]=true;
            if(match[v]==-1||Dfs(match[v])){
                match[v]=u;
                return true;
            }
        }
    }
    return false;
}
int Hungary(){
    memset(match,-1,sizeof(match));
    int ret=0;
    for(int i=1;i<=N;i++){
        if(A[i]&&B[i]) continue;
        memset(vis,false,sizeof(vis));
        if(Dfs(i)) ret++;
    }
    return ret;
}
int main(){
    int Test=readint();
    while(Test--){
        memset(fir,-1,sizeof(fir));
        cnt=0;
        N=readint();
        for(int i=1;i<=N;i++) A[i]=readint();
        for(int i=1;i<=N;i++) B[i]=readint();
        for(int i=1;i<=N;i++)
            for(int j=1;j<=N;j++){
                bool x=readint();
                if(!x||(!A[j])||(A[i]&&B[i])) continue;
                Add(i,j+N);
            }
        for(int i=1;i<=N;i++)
            if(A[i]&&(!B[i]))
                Add(i,i+N);
        int tot=N;
        for(int i=1;i<=N;i++)
            if(A[i]&&B[i])
                tot--;
        int Ans=Hungary();
        if(Ans<tot) printf("%c%c%c
",84,95,84);
        else printf("%c%c%c
",94,95,94);
    }
    return 0;
}