【网络流24题】圆桌问题

（网络流24题大多需要spj，所以需要一个有spj的oj，本系列代码均在www.oj.swust.edu.cn测试通过）
这道题的模型应该是有一种专业名称的，但是那都没用，思路才是最关键的。
我们从源点向每一个公司连接一条容量为公司人数的边，然后每个公司向所有餐桌连一条容量为1得边，每个餐桌向汇点连一条容量为餐桌人数上限的边，这样跑一边最大流，如果最大流等于所有公司的总人数之和，证明有解，反之无解

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<ctime>
#include<cstring>
#include<string>
#include<iomanip>
#include<iostream>
#include<algorithm>
using namespace std;
#define INF 100000000
struct bian
{
    int l,r,f,lei;
}a[1000000];
int s=0,t=99999;
int d[100000];
int fir[1000000];
int nex[1000000];
int wz[1000000];
bool bfs()
{
    static int dui[100000];
    memset(d,-1,sizeof(d));
    int top=1,my_final=2;
    dui[1]=s;
    d[s]=1;
    while(top<my_final)
    {
        int u=dui[top++];
        for(int o=fir[u];o;o=nex[o])
        {
            if(a[o].f && d[a[o].r]==-1)
            {
                d[a[o].r]=d[u]+1;
                dui[my_final++]=a[o].r;
                if(a[o].r==t) return true;
            }
        }
    }
    return false;
}
int dinic(int u,int flow)
{
    if(u==t) return flow;
    int left=flow;
    for(int o=fir[u];o&&left;o=nex[o])
    {
        if(a[o].f && d[a[o].r]==d[u]+1)
        {
            int temp=dinic(a[o].r,min(a[o].f,left));
            if(!temp) d[a[o].r]=-1;
            left-=temp;
            a[o].f-=temp;
            a[o^1].f+=temp;
        }
    }
    return flow-left;
}
int tot=1;
void add_edge(int l,int r,int f,int lei=0)
{
    a[++tot].l=l;
    a[tot].r=r;
    a[tot].f=f;
    a[tot].lei=lei;
    nex[tot]=fir[l];
    fir[l]=tot;
}
int main()
{
    int n,m;
    scanf("%d%d",&m,&n);
    int zong=0;
    for(int i=1;i<=m;i++)
    {
        int x;
        scanf("%d",&x);
        zong+=x;
        add_edge(s,i,x);
        add_edge(i,s,0);
    }
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        add_edge(m+i,t,x);
        add_edge(t,m+i,0);
    }
    for(int i=1;i<=m;i++)
    {
        wz[i]=tot+1;
        for(int j=1;j<=n;j++)
        {
            add_edge(i,m+j,1,1);
            add_edge(m+j,i,0);
        }
    }
    wz[m+1]=tot+1;
    int ans=0;
    while(bfs()) ans+=dinic(s,INF);
    if(ans!=zong) {cout<<0;return 0;}
    else
    {
        cout<<1<<endl;
        for(int i=1;i<=m;i++)
        {
            for(int j=wz[i];j<wz[i+1];j++) if(a[j].f==0 && a[j].lei) printf("%d ",a[j].r-m);
            printf("\n");
        }
    }
    return 0;
}