bzoj1433 [ZJOI2009]假期的宿舍

题目

再次根据数据猜算法，233，网络流跑二分图匹配吧。

每个人需要一张床，所以就是床与人匹配。
然后根据题目描述来建边。

#include<bits/stdc++.h>
#define N 150
#define M 3000
#define inf ~0u>>1
using namespace std;
int n,cnt,T,S,bed[N+1],x,ans,Case;
int first[N+1],nex[M+1],to[M+1],fl[M+1],siz,dis[N+1];
inline char nc()
{
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    int x=0,b=1;
    char c=nc();
    for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
    for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
    return x*b;
}
inline void add(int x,int y,int z)
{
    nex[siz]=first[x];
    first[x]=siz;
    to[siz]=y;
    fl[siz]=z;
    siz++;
}
inline void insert(int x,int y,int z)
{
    add(x,y,z),add(y,x,0);
}
inline bool BFS(int s,int t)
{
    queue <int> Q;
    for(int i=S;i<=T;i++)dis[i]=inf;
    dis[s]=0,Q.push(s);
    while(!Q.empty())
    {
        int x=Q.front();Q.pop();
        for(int i=first[x];i!=-1;i=nex[i])
        {
            int u=to[i];
            if(fl[i]&&dis[x]+1<dis[u])
            {
                dis[u]=dis[x]+1;
                if(u==t)return true;
                Q.push(u);
            }
        }
    }
    if(dis[t]==inf)return false;
    return true;
}
inline int DFS(int s,int f,int t)
{
    if(s==t)return f;
    int rest=f;
    for(int i=first[s];i!=-1;i=nex[i])
    {
        int u=to[i];
        if(fl[i]&&rest&&dis[u]==dis[s]+1)
        {
            int tmp=DFS(u,min(fl[i],rest),t);
            if(tmp==0)dis[u]=0;
            fl[i]-=tmp;
            fl[i^1]+=tmp;
            rest-=tmp;
        }
    }
    return f-rest;
}
inline int Dinic(int s,int t)
{
    int ans=0;
    while(BFS(s,t))ans+=DFS(s,inf,t);
    return ans;
}
int main()
{
    //freopen("in.txt","r",stdin);
    Case=read();
    while(Case--)
    {
        cnt=0,siz=0;
        memset(first,-1,sizeof(first));
        n=read();
        S=0,T=2*n+1;
        for(int i=1;i<=n;i++)bed[i]=read();
        for(int i=1;i<=n;i++)if(bed[i])insert(i+n,T,1);
        for(int i=1;i<=n;i++)
        {
            x=read();
            if((!x&&bed[i])||!bed[i])insert(S,i,1),cnt++;
        }
        for(int i=1;i<=n;i++)
            for(int j=1;j<=n;j++)
            {
                x=read();
                if(x||i==j)insert(i,j+n,1);
            }
        ans=Dinic(S,T);
        if(ans==cnt)puts("^_^");
        else puts("T_T");
    }
    return 0;
}

36ms，然而他们28ms如何写出来的啊233。