題目描述

假設有來自m 個不同單位的代表參加一次國際會議。每個單位的代表數分別為ri (i =1,2,……,m)。

會議餐廳共有n 張餐桌，每張餐桌可容納ci (i =1,2,……,n)個代表就餐。

為了使代表們充分交流，希望從同一個單位來的代表不在同一個餐桌就餐。試設計一個演算法，給出滿足要求的代表就餐方案。

對於給定的代表數和餐桌數以及餐桌容量，程式設計計算滿足要求的代表就餐方案。

輸入輸出格式

輸入格式：

第1 行有2 個正整數m 和n，m 表示單位數，n 表示餐桌數，1<=m<=150, 1<=n<=270。

第2 行有m 個正整數，分別表示每個單位的代表數。

第3 行有n 個正整數，分別表示每個餐桌的容量。

輸出格式：

如果問題有解，第1 行輸出1，否則輸出0。接下來的m 行給出每個單位代表的就餐桌號。如果有多個滿足要求的方案，只要輸出1 個方案。

輸入輸出樣例

輸入樣例#1： 複製

4 5
4 5 3 5
3 5 2 6 4

輸出樣例#1： 複製

1
1 2 4 5
1 2 3 4 5
2 4 5
1 2 3 4 5

解題思路：題目資訊已經很明確了，代表向原點連邊，容量為人數，因為同一地區的代表不能坐在同一個桌子，所以每個點向桌子連邊容量為1，然後每個桌子向原點連邊，容量為桌子的容量。

如果最大流為總共的人數的話，該方案就是可行的，否則的話就是不可行的。

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<set>
#include<cmath>
#include<map>
#include<algorithm>
#include<cstring>
using namespace std;
#define rep(i,j,k) for(int i=j;i<=k;i++)
#define sca(x) scanf("%d",&x)
#define pb(x) push_back(x)
#define per(i,j,k) for(int i=j;i>=k;i--)
#define inf 0x3f3f3f3f
#define LL long long
#define N 5005
#define MAXN 100000
#define inf 0x3f3f3f3f

struct edg
{
    int to,v,w,nt;
}g[MAXN];
int tot,S,T;
int head[N],a[N],b[N];
void addedg(int u,int v,int w)
{
    g[tot].to=v;
    g[tot].w=w;
    g[tot].nt=head[u];
    head[u]=tot++;

    g[tot].to=u;
    g[tot].w=0;
    g[tot].nt=head[v];
    head[v]=tot++;

}

int dep[N+10],cur[N];
bool bfs(int s,int t)
{
    memset(dep,-1,sizeof(dep));
    queue<int>q;
    q.push(s);
    dep[s]=0;
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=head[u];i!=-1;i=g[i].nt)
        {
            int to=g[i].to;
            if(dep[to]<0&&g[i].w>0)
            {
                dep[to]=dep[u]+1;
                q.push(to);
            }
        }
    }
    return dep[t]!=-1;
}

int dfs(int s,int flow)
{
    int tmp=0;
    if(s==T)return flow;
    for(int& i=cur[s];i!=-1;i=g[i].nt)
    {
        int to=g[i].to;
        if(dep[to]==dep[s]+1&&g[i].w>0&&
        (tmp=dfs(to,min(flow,g[i].w))))
        {
            g[i].w-=tmp;
            g[i^1].w+=tmp;
            return tmp;
        }
    }
    return 0;
}

int dinic(int s,int t)
{
    int d,ans=0;
    while(bfs(s,t))
    {
        rep(i,0,t)cur[i]=head[i];
        while(d=dfs(s,inf))ans+=d;
    }
    return ans;
}

void P(int n)
{
    vector<int>v[200];
    rep(i,1,n)
    {
        for(int j=head[i];j!=-1;j=g[j].nt)
        {
            int to=g[j].to;
            if(g[j].w==0)
            {
                v[i].pb(to-n);
            }
        }
    }
    rep(i,1,n)
    {
        int len=v[i].size();
        rep(j,0,len-1)
        {
            printf("%d ",v[i][j]);
        }
        puts("");
    }
}
int main()
{
    int n,m;
    sca(m),sca(n);
    memset(head,-1,sizeof(head));
    int sum=0;
    rep(i,1,m)sca(a[i]),sum+=a[i];
    rep(i,1,n)sca(b[i]);
    S=0,T=n+m+1;
    rep(i,1,m)addedg(S,i,a[i]);
    rep(i,1,n)addedg(i+m,T,b[i]);
    rep(i,1,m)
    {
        rep(j,1,n)
        {
            addedg(i,j+m,1);
        }
    }
    int ans=dinic(S,T);
    if(ans==sum)
    {
        printf("1\n");
        P(m);
    }
    else printf("0\n");
}