「网络流 24 题」圆桌聚餐

网络流水题,详细看代码

#include<bits/stdc++.h>
using namespace std;

int n,m,tot=-1,h[3005],ans=0,sum=0;
struct node{
    int from,next,to,rest,full;
    int last;
}e[100005];
void add(int x,int y,int z){
    tot++;
    e[tot].next=h[x];
    h[x]=tot;
    e[tot].from=x;
    e[tot].to=y;
    e[tot].rest=z;
    e[tot].full=z;
}

int dis[3005],g[3005],flow[3005];
bool vis[3005];

int bfs(int s,int t){
    queue<int>q;
    dis[s]=0;
    q.push(s);vis[s]=true;
    while(!q.empty()){
        int u=q.front();vis[u]=false;q.pop();
        for(int i=h[u];i!=(-1);i=e[i].next){
            if(dis[e[i].to]>dis[u]+1&&g[e[i].to]==(-1)&&e[i].rest>0){
                g[e[i].to]=i;
                flow[e[i].to]=min(flow[u],e[i].rest);
                dis[e[i].to]=dis[u]+1;
                if(vis[e[i].to]==false){
                    vis[e[i].to]=true;
                    q.push(e[i].to);
                }
            }
        }
    }
}

int EK(int s,int t){
    while(1){
        memset(vis,false,sizeof(vis));
        memset(dis,0x7f,sizeof(dis));
        memset(flow,0x7f,sizeof(flow));
        memset(g,-1,sizeof(g));
        bfs(s,t);
        if(g[t]==(-1))return 0;
        ans+=flow[t];
        for(int p=t;p!=(s);p=e[g[p]].from){
            e[g[p]].rest-=flow[t];
            e[g[p]^1].rest+=flow[t];
        }    
        
    }
}

int main(){
    memset(h,-1,sizeof(h));
    cin>>m>>n;
    for(int i=1;i<=m;i++){
        int co;cin>>co;sum+=co;
        add(0,i,co);
        add(i,0,0);
        for(int j=1;j<=n;j++){
            add(i,j+m,1);
            add(j+m,i,0);
        }
    }
    for(int i=1;i<=n;i++){
        int co;cin>>co;
        add(i+m,n+m+1,co);
        add(n+m+1,i+m,0);
    }
    EK(0,n+m+1);
    if(sum>ans){
        cout<<0<<endl;
        exit(0);
    }
    cout<<1<<endl;
    for(int i=1;i<=m;i++){
        for(int j=h[i];j!=(-1);j=e[j].next){
            if(e[j].to!=0&&e[j].rest==0){
                cout<<e[j].to-m<<" ";
            }
        }
        cout<<endl;
    }
}