洛谷 P2055 [ZJOI2009]假期的宿舍（最大流）

传送门

---

很明显是二分图最大匹配，不过似乎网络流更快，所以这里讲网络流吧。
每张床向汇点连流量为1的边，源点向每个在学校住的在校学生或其他学校的学生连流量为1的边，每个要在学校住的学生向自己的床与自己认识的人的床连边，然后跑最大流。如果最大流=要在学校住的学生人数，就表示存在方案，反之则不存在。

Code:

#include<cstdio>
#include<cstdlib>
#include<cstring>

struct node{int x,y,c,next,other;}a[5210];
int n,t,len=0,st,ed,ans=0,tot=0,first[1010],h[1010],q[1010],sch[1010],stay[1010];

void ins(int x,int y,int c)
{
    a[++len]=((node){x,y,c,first[x],len+1});first[x]=len;
    a[++len]=((node){y,x,0,first[y],len-1});first[y]=len;
}

bool bfs()
{
    int head=1,tail=2;q[1]=st;
    memset(h,0,sizeof(h));h[st]=1;
    while(head!=tail)
    {
        int x=q[head];
        for(int i=first[x];i;i=a[i].next)
        {
            int y=a[i].y;
            if(h[y]==0 && a[i].c>0)
            {
                q[tail++]=y;
                h[y]=h[x]+1;
            }
        }
        head++;
    }
    if(h[ed]==0) return false;
    return true;
}

int mymin(int x,int y)
{
    return x<y?x:y;
} 

int dfs(int x,int flow)
{
    if(x==ed) return flow;
    int tt=0,minf=0;
    for(int i=first[x];i;i=a[i].next)
    {
        int y=a[i].y;
        if(a[i].c>0 && h[y]==(h[x]+1) && tt<flow)
        {
            minf=dfs(y,mymin(flow-tt,a[i].c));
            tt+=minf;
            a[i].c-=minf;a[a[i].other].c+=minf;
        }
    }
    if(tt==0) h[x]=0;
    return tt;
}

int main()
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        st=2*n+1;ed=st+1;tot=0;ans=0;len=0;
        memset(first,0,sizeof(first));
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&sch[i]);
            if(sch[i]==1) ins(i+n,ed,1);
        }
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&stay[i]);
            if((stay[i]==0 && sch[i]==1) || sch[i]==0) tot++,ins(st,i,1);
        }
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=n;j++)
            {
                int x;scanf("%d",&x);
                if(x==1 || i==j) ins(i,j+n,1);
            }
        }
        while(bfs()) ans+=dfs(st,0x3f3f3f3f);
        if(ans==tot) printf("^_^\n");
        else printf("T_T\n");
    }
}