【网络流24题】圆桌问题

Description

假设有来自 n个不同单位的代表参加一次国际会议。每个单位的代表数分别为
ri ,i =1,2,.,n 。会议餐厅共有 m张餐桌，每张餐桌可容纳 ci (i =1,2,., m) 个代表就餐。为了使代表们充分交流，希望从同一个单位来的代表不在同一个餐桌就餐。试设计一个算法，给出满足要求的代表就餐方案。

对于给定的代表数和餐桌数以及餐桌容量，编程计算满足要求的代表就餐方案。

Input

文件第 1行有 2个正整数 m和 n，m表示单位数，n表示餐桌数， 1<=m<=150, 1<=n<=270。文件第 2行有 m个正整数，分别表示每个单位的代表数。文件第 3行有 n个正整数，分别表示每个餐桌的容量。

Output

如果问题有解，在文件第 1行输出 1，否则输出 0。接下来的 m行给出每个单位代表的就餐桌号。如果有多个满足要求的方案，只要输出 1个方案。

Sample Input

4 5
4 5 3 5
3 5 2 6 4

Sample Output

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

题解

网络流最大流。
1.从S向每个$r_i$顶点连接一条流量为该单位代表人数的有向边。
2.从每个$c_i$顶点到T连接一条流量为该餐桌可容纳人数的有向边。
3.从每个$r_i$顶点到每个$c_i$顶点连接一条流量为1的有向边。

求最大流，如果所有单位的代表人数之和等于最大流量，则有解，否则无解。

code

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define N 100010
#define T 100000
#define inf 0x7fffffff
using namespace std;
struct node
{
    int next,to,s;
};
node Edge[N<<1];
int n,m,ans,tot=1;
int head[N];
int h[N];

void add(int x,int y,int z)
{
    Edge[++tot].next=head[x];
    Edge[tot].to=y;
    Edge[tot].s=z;
    head[x]=tot;
}

void ins(int x,int y,int z)
{
    add(x,y,z),add(y,x,0);
}

bool bfs()
{
    memset(h,-1,sizeof(h));
    queue<int>Q;
    Q.push(0);
    h[0]=0;
    while(!Q.empty())
    {
        int now=Q.front();
        Q.pop();
        for(int i=head[now];i;i=Edge[i].next)
        {
            int to=Edge[i].to,s=Edge[i].s;
            if(!s||h[to]!=-1)continue;
            h[to]=h[now]+1;
            Q.push(to);
        }
    }
    if(h[T]==-1)return false;
    return true;
}

int dfs(int u,int f)
{
    if(u==T)return f;
    int used=0;
    for(int i=head[u];i;i=Edge[i].next)
    {
        int to=Edge[i].to,s=Edge[i].s;
        if(!s||h[to]!=h[u]+1)continue;
        int w=f-used;
        w=dfs(to,min(w,s));
        used+=w;
        Edge[i].s-=w;
        Edge[i^1].s+=w;
        if(used==f)return used;
    }
    if(!used)h[u]=-1;
    return used;
}

void dinic()
{
    while(bfs())
        ans-=dfs(0,inf);
}

void getans()
{
    for(int i=1;i<=n;i++)
    {
        for(int j=head[i];j;j=Edge[j].next)
        {
            if(Edge[j].s||!Edge[j].to)continue;
            printf("%d ",Edge[j].to-n);
        }
        puts("");
    }
}

int main()
{
    cin>>n>>m;
    for(int x,i=1;i<=n;i++)
    {
        scanf("%d",&x);
        ins(0,i,x);
        ans+=x;
    }
    for(int x,i=1;i<=m;i++)
    {
        scanf("%d",&x);
        ins(i+n,T,x);
    }
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            ins(i,j+n,1);
    dinic();
    if(ans){puts("0");return 0;}
    else puts("1");
    getans();
}