【网络流24题】圆桌聚餐（二分图）

传送门

    圆桌聚餐

I think

    增设源汇点S T，S向单位连容量为单位人数的边，每张桌子向T连容量为桌子容纳人数的边，每个人向每张桌子连容量为1的边，整张图的最大流==总人数时即有解。

Code

#include<cstdio>
#include<queue>
using namespace std;

const int sm = 450+5;
const int sn = 82000;
const int Inf = 0x3f3f3f3f;

int M,N,S,T,tot=1,Flw,sum; bool fl;
int to[sn],nxt[sn],hd[sm],c[sn];
int lev[sm],cur[sm];

int Min(int x,int y) { return x<y?x:y; } 
void Add(int u,int v,int w) {
    to[++tot]=v,nxt[tot]=hd[u],hd[u]=tot,c[tot]=w;
    to[++tot]=u,nxt[tot]=hd[v],hd[v]=tot,c[tot]=0;
}
bool Bfs() {
    queue<int>q; int t;
    for(int i=1;i<=T;++i) lev[i]=0;
    lev[S]=1,q.push(S);
    while(!q.empty()) {
        t=q.front(),q.pop();
        for(int i=hd[t];i;i=nxt[i])
            if(c[i]&&!lev[to[i]]) {
                lev[to[i]]=lev[t]+1;
                if(to[i]==T) return 1;
                q.push(to[i]);
            }
    }
    return 0;
}
int Dfs(int x,int mx) {
    if(x==T||!mx) return mx;
    int f;
    for(int i=cur[x]?cur[x]:hd[x];i;i=nxt[i]) {
        cur[x]=i;
        if(c[i]>0&&lev[to[i]]==lev[x]+1) {
            if(f=Dfs(to[i],Min(mx,c[i])))
                return c[i]-=f,c[i^1]+=f,f;
        }
    }
    return 0;
}
void Dinic() {
    int f;
    while(Bfs()) {
        for(int i=1;i<=T;++i) cur[i]=0;
        while(f=Dfs(S,Inf))Flw+=f;
    } 
}       

int main() {
    int u;
    scanf("%d%d",&M,&N);
    S=M+N+1,T=S+1;
    for(int i=1;i<=M;++i) {
        scanf("%d",&u);
        sum+=u;
        if(u>N) fl=1;
        if(!fl) Add(S,i,u);
    }
    for(int i=1;i<=N;++i) {
        scanf("%d",&u);
        if(!fl) Add(M+i,T,u);
    }
    if(fl) { puts("0");return 0; }
    for(int i=1;i<=M;++i)
        for(int j=M+1;j<=M+N;++j)
            Add(i,j,1);
    Dinic();
    if(Flw<sum) puts("0");
    else {
        puts("1");
        for(int i=1;i<=M;++i) {
            for(int j=hd[i];j;j=nxt[j])
                if(to[j]!=S&&!c[j]) printf("%d ",to[j]-M);
            putchar(10);
        }
    }
    return 0;
}