洛谷P3254 圆桌问题（最大流）

传送门

 

一道良心啊……没那么多麻烦了……

从$S$向所有单位连边，容量为单位人数，从所有桌子向$T$连边，容量为桌子能坐的人数，从每一个单位向所有桌子连边，容量为$1$，然后跑一个最大流，看一看$S$到单位这一边流满了没，如果没有就无解。方案的话，就看单位到哪一个桌子有流就行

因为手写队列然后两个$t$弄混了结果死都不知道哪里错……

//minamoto
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define inf 0x3f3f3f3f
using namespace std;
#define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
char buf[1<<21],*p1=buf,*p2=buf;
inline int read(){
    #define num ch-'0'
    char ch;bool flag=0;int res;
    while(!isdigit(ch=getc()))
    (ch=='-')&&(flag=true);
    for(res=num;isdigit(ch=getc());res=res*10+num);
    (flag)&&(res=-res);
    #undef num
    return res;
}
char sr[1<<21],z[20];int C=-1,Z;
inline void Ot(){fwrite(sr,1,C+1,stdout),C=-1;}
inline void print(int x){
    if(C>1<<20)Ot();
    while(z[++Z]=x%10+48,x/=10);
    while(sr[++C]=z[Z],--Z);
}
const int N=505,M=100005;
int head[N],Next[M],ver[M],edge[M],cur[N],tot=1;
int dep[N],n,m,u,v,e,vis[N],s,t,sum=0;
queue<int> q;
inline void add(int u,int v,int e){
    ver[++tot]=v,Next[tot]=head[u],head[u]=tot,edge[tot]=e;
    ver[++tot]=u,Next[tot]=head[v],head[v]=tot,edge[tot]=0;
}
bool bfs(){
    memset(dep,-1,sizeof(dep));
    while(!q.empty()) q.pop();
    for(int i=0;i<=n+m+1;++i) cur[i]=head[i];
    q.push(s),dep[s]=0;
    while(!q.empty()){
        int u=q.front();q.pop();
        for(int i=head[u];i;i=Next[i]){
            int v=ver[i];
            if(dep[v]<0&&edge[i])
            dep[v]=dep[u]+1,q.push(v);
        }
    }
    return ~dep[t];
}
int dfs(int u,int limit){
    if(!limit||u==t) return limit;
    int flow=0,f;
    for(int i=cur[u];i;i=Next[i]){
        int v=ver[i];cur[u]=i;
        if(dep[v]==dep[u]+1&&(f=dfs(v,min(limit,edge[i])))){
            flow+=f,limit-=f;
            edge[i]-=f,edge[i^1]+=f;
            if(!limit) break;
        }
    }
    return flow;
}
int dinic(){
    int flow=0;
    while(bfs()) flow+=dfs(s,inf);
    return flow;
}
int main(){
    m=read(),n=read();
    s=0,t=n+m+1;
    for(int i=1;i<=m;++i){
        e=read(),add(s,i,e),sum+=e;
        for(int j=1;j<=n;++j)
        add(i,j+m,1);
    }
    for(int i=1;i<=n;++i){
        e=read();add(i+m,t,e);
    }
    if(dinic()!=sum) return puts("0"),0;
    print(1),sr[++C]='\n';
    for(int i=1;i<=m;++i){
        for(int j=head[i];j;j=Next[j]){
            if(!edge[j]&&ver[j]>m&&ver[j]<=n+m)
            print(ver[j]-m),sr[++C]=' ';
        }
        sr[++C]='\n';
    }
    Ot();
    return 0;
}