[ZJOI2009]BZOJ1433 假期的宿舍

传送门

题解：

SB二分图匹配，我会说我提交了五六次么QwQ

注意清零和i和i+n（人和本人的床）连边

。。

#include<queue>
#include<cstdio>
#include<climits>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 110
#define MAXM 20000
#define INF INT_MAX
#define build_edge(u,v,c) add_edge(u,v,c),add_edge(v,u,0)
#define P(x,y) (~-(x)*m+(y))
#define inrange(x,y) (1<=x&&x<=n&&1<=y&&y<=m)
using namespace std;
struct edges{
    int to,pre,resf;
}e[MAXM];
int etop,h[MAXN],lev[MAXN],cur[MAXN];
bool vis[MAXN];queue<int> q;int totans;
int dx[]={0,-1,0,1,0},
    dy[]={0,0,1,0,-1};
inline int add_edge(int u,int v,int c)
{
    etop++;
    e[etop].to=v;
    e[etop].pre=h[u];
    h[u]=etop;
    e[etop].resf=c;
    return 0;
}
inline bool bfs(int s,int t)
{
    memset(vis,false,sizeof(vis));
    memset(lev,0,sizeof(lev));
    while(!q.empty()) q.pop();
    q.push(s);lev[s]=0;vis[s]=true;
    while(!q.empty())
    {
        int x=q.front();q.pop();
        for(int i=h[x];i;i=e[i].pre)
            if(!vis[e[i].to]&&e[i].resf)
            {
                vis[e[i].to]=true;
                lev[e[i].to]=lev[x]+1;
                q.push(e[i].to);
            }
    }
    return vis[t];
}
int dfs(int s,int t,int a)
{
    if(s==t||!a) return a;
    int flow=0,f;
    for(int &i=cur[s];i;i=e[i].pre)
        if(lev[e[i].to]==lev[s]+1&&(f=dfs(e[i].to,t,min(a,e[i].resf)))>0)
        {
            a-=f;flow+=f;e[i].resf-=f;
            e[((i-1)^1)+1].resf+=f;
            if(!a) break;
        }
    return flow;
}
int is_stu[MAXN];
int main()
{
    int T;scanf("%d",&T);
    while(T--)
    {
        memset(h,0,sizeof(h));
        int n,s,t;scanf("%d",&n);
        t=2*n+1;s=etop=totans=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&is_stu[i]);
            if(is_stu[i]) build_edge(i+n,t,1);
            else build_edge(s,i,1),totans++;
        }
        for(int i=1;i<=n;i++)
            if(is_stu[i]) build_edge(i,i+n,1);
        for(int i=1;i<=n;i++)
        {
            int x;scanf("%d",&x);
            if(is_stu[i]&&!x)
                build_edge(s,i,1),totans++;
        }
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
            {
                int x;scanf("%d",&x);
                if(x&&is_stu[j]) build_edge(i,j+n,1);
            }
        int flow=0;
        while(bfs(s,t))
        {
            for(int i=s;i<=t;i++)
                cur[i]=h[i];
            flow+=dfs(s,t,INF);
        }
        if(flow==totans) printf("^_^\n");
        else printf("T_T\n");
    }
    return 0;
}