网络流24题 05圆桌聚餐

题目描述
假设有来自 n n 个不同单位的代表参加一次国际会议。每个单位的代表数分别为 r_i r
​i
​​ 。会议餐厅共有 m m 张餐桌，每张餐桌可容纳 c_i c
​i
​​ 个代表就餐。
为了使代表们充分交流，希望从同一个单位来的代表不在同一个餐桌就餐。

试设计一个算法，给出满足要求的代表就餐方案。
输入格式
文件第 1 1 行有 2 2 个正整数 m m 和 n n， m m 表示单位数， n n 表示餐桌数。
文件第 2 2 行有 m m 个正整数，分别表示每个单位的代表数。
文件第 3 3 行有 n n 个正整数，分别表示每个餐桌的容量。
输出格式
如果问题有解，在文件第 1 1 行输出 1 1，否则输出 0 0。
接下来的 m m 行给出每个单位代表的就餐桌号。如果有多个满足要求的方案，只要输出一个方案。
样例
样例输入

4 5
4 5 3 5
3 5 2 6 4
样例输出

1
1 2 4 5
1 2 3 4 5
2 4 5
1 2 3 4 5
数据范围与提示
1≤m≤150,1≤n≤270

一开始想的是直接人与人之间连边，不同单位的互相连边，但是明显会炸。。
然后就有些蒙蔽了，，事实上这说明我对于网络流本质理解还是不深刻。。

所有单位向超级源连容量为人数的边，所有桌子向超级汇连容量为桌子容量的边，然后单位和桌子之间连大小为1，这是因为为了保证最优，每个单位在一个桌子上只能有一个人。这样跑出来最大流，如果结果不为总人数，则明显无解。
方案则是流完容量的边的所有到达点。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define S 0
#define T 1001
using namespace std;
const int N=3e5+5;
const int M=2e3+5;
const int inf=1e9;
int head[N],go[N],next[N],val[N],tot=1;
int n,m;
int q[N],dis[M],ans;
inline bool bfs()
{
    int t=0,w=1;
    //int w,used=0;
    memset(dis,-1,sizeof(dis));
    dis[0]=0;
    q[1]=0;
    while (t<w)
    {
        int x=q[++t];
        for(int i=head[x];i;i=next[i])
        {
            int v=go[i];
            if(val[i]&&dis[v]==-1)
            {
                dis[v]=dis[x]+1;
                q[++w]=v;
            }
        }
    }
    return dis[T]!=-1;
}
inline int dfs(int x,int f)
{
    if (x==T)return f;
    int w,used=0;
    for(int i=head[x];i;i=next[i])
    {
        int v=go[i];
        if (val[i]&&dis[v]==dis[x]+1)
        {
            w=f-used;
            w=dfs(v,min(w,val[i]));
            val[i]-=w;
            val[i^1]+=w;
            used+=w;
            if (used==f)return f;
        }
    }
    if (!used)dis[x]=-1;
    return used;
}
inline void dinic()
{
    while(bfs())ans-=dfs(0,inf);
}
inline void add(int x,int y,int z)
{
    go[++tot]=y;
    val[tot]=z;
    next[tot]=head[x];
    head[x]=tot;
}
inline void ins(int x,int y,int z)
{
    add(x,y,z);
    add(y,x,0);
}
int main()
{
    scanf("%d%d",&m,&n);
    fo(i,1,m)
    {
        int x;
        scanf("%d",&x);
        ans+=x;
        ins(0,i,x);
    }
    fo(i,1,n)
    {
        int x;
        scanf("%d",&x);
        ins(i+m,T,x);
    }
    fo(i,1,m)
    fo(j,1,n)
    ins(i,j+m,1);
    //printf("%d\n",ans);
    dinic();
    //printf("%d\n",ans);
    if (ans)
    {
        printf("0\n");
        return 0;
    }
    printf("1\n");
    fo(i,1,m)
    {
        for(int j=head[i];j;j=next[j])
        {
            if (!val[j])printf("%d ",go[j]-m);
        }
            printf("\n");
    }
    return 0;
}