圆桌问题

和试题库问题很类似
套路满满

#include <cstdio>
#include <cstring>
#include <queue>
#include <iostream>
using namespace std;
const int S=0,T=504,inf=0x3f3f3f3f;
 
int n,m,sum;
struct Edge {
    int to,nxt,val;
} e[2000005];
int mx;
int maxflow;
int cur[20005];
int head[20005],cnt=1;

void add(int bg,int ed,int val) {
    e[++cnt].to=ed;
    e[cnt].nxt=head[bg];
    e[cnt].val=val;
    head[bg]=cnt;
}
int h[20005];
queue<int>q;
bool bfs() {
    memset(h,-1,sizeof h);
    q.push(S);
    h[S]=0;
    while(!q.empty()) {
        int x=q.front();
        q.pop();
        for(int i=head[x]; i; i=e[i].nxt) {
            if(e[i].val&&h[e[i].to]==-1) {
                h[e[i].to]=h[x]+1;
                q.push(e[i].to);
            }
        }
    }
    return h[T]!=-1;
}
int dfs(int x,int f) {
    if(x==T)return f;
    int tp,used=0;
    for(int i=cur[x]; i; i=e[i].nxt) {
        if(e[i].val&&h[e[i].to]==h[x]+1) {
            tp=dfs(e[i].to,min(e[i].val,f-used));
            e[i].val-=tp;
            if(e[i].val)cur[x]=i;
            e[i^1].val+=tp;
            used+=tp;
            if(used==f)return f;
        }
    }
    if(!used)h[x]=-1;
    return used;
}
void dinic() {
    while(bfs()) {
        memcpy(cur,head,sizeof head);
        maxflow+=dfs(S,inf);
    }
}
void insert(int a,int b,int c){add(a,b,c);add(b,a,0);}
int main(){
	scanf("%d%d",&m,&n);
	for(int i=1,x;i<=m;i++) {scanf("%d",&x);insert(S,i,x);sum+=x;}
	for(int i=1,x;i<=n;i++) {scanf("%d",&x);insert(i+m,T,x);}
	for(int i=1;i<=m;i++) for(int j=1;j<=n;j++) insert(i,j+m,1);
	dinic();
	if(maxflow==sum){
		printf("1
");
		for(int i=1;i<=m;i++) {
			for(int j=head[i];j;j=e[j].nxt){
				if(!e[j].val&&e[j].to>m) printf("%d ",e[j].to-m);
			}
			printf("
");
		}
	}
	else puts("0");
}