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

来源：互联网发布：淘宝怎么查看购物车编辑：程序博客网时间：2024/06/04 20:07

题面：BZOJ1433 洛谷2055
这个题预处理好麻烦啊。。。题面也很绕
先说预处理吧，我们首先转成图，也就是把人和床分开变成二分图
具体怎么处理输入呢？
首先在校不回家的学生向自己的床连一条边，
然后所有人向认识的人里面有床的人的床连边
注意回家的学生要忽略。。。
最后每张床连向超级汇点，超级原点向每个在学校的人（包括学生和朋友）连边
然后就跑二分图匹配啦~

#include<bits/stdc++.h>using namespace std;int dist[100001],nedge=-1,head[100001],nex[100001],p[100001],c[100001],m,n,s,t,x,y,z;int a[100001],b[100001];inline void addedge(int x,int y,int z){    p[++nedge]=y;c[nedge]=z;    nex[nedge]=head[x];head[x]=nedge;}bool bfs(int s,int e){    int now;queue<int>q;    memset(dist,-1,sizeof dist);    dist[s]=1;q.push(s);    while(!q.empty()){        now=q.front();q.pop();        for(int k=head[now];k>-1;k=nex[k]){            if(dist[p[k]]<0&&c[k]>0){                dist[p[k]]=dist[now]+1;                q.push(p[k]);            }        }    }    if(dist[e]>0)return 1;    return 0;}int finds(int x,int low){    int a;    if(x==t)return low;    int used=0;    for(int k=head[x];k>-1;k=nex[k]){        if(c[k]&&dist[x]+1==dist[p[k]]&&(a=finds(p[k],min(low-used,c[k])))){            c[k]-=a;            c[k^1]+=a;            used+=a;        }        if(used==low)break;    }    if(used==0)dist[x]=-1;    return used;}inline int dinic(int s,int t){    int flow=0;    while(bfs(s,t))flow+=finds(s,0x3f3f3f3f);    return flow;}int main(){    int T;scanf("%d",&T);    while(T--){        memset(p,0,sizeof p);memset(c,0,sizeof c);        memset(nex,-1,sizeof nex);memset(head,-1,sizeof head);        scanf("%d",&n);int k=0,d=0;s=0;t=2*n+1;        for(int i=1;i<=n;i++)scanf("%d",&a[i]);        for(int i=1;i<=n;i++){            scanf("%d",&b[i]);            if(!a[i])k++,addedge(s,i,1),addedge(i,s,0);            if(a[i]&&!b[i])k++,addedge(s,i,1),addedge(i,s,0),addedge(i,i+n,1),addedge(i+n,i,0);            if(a[i])addedge(i+n,t,1),addedge(t,i+n,0);        }        for(int i=1;i<=n;i++)            for(int j=1;j<=n;j++){                int x;scanf("%d",&x);                if(x==1&&a[j])addedge(i,j+n,1),addedge(j+n,i,0);            }        if(dinic(s,t)==k)puts("^_^");        else puts("T_T");    }    return 0;}

1 0