loj #6004. 「网络流 24 题」圆桌聚餐(最大流)

#6004. 「网络流 24 题」圆桌聚餐

内存限制：256 MiB 时间限制：5000 ms 标准输入输出

题目类型：传统 评测方式：Special Judge

上传者： 匿名

提交 提交记录 统计 讨论 测试数据

题目描述

假设有来自 $n$ 个不同单位的代表参加一次国际会议。每个单位的代表数分别为 ri r_ir​i​​。会议餐厅共有 $m$ 张餐桌，每张餐桌可容纳 ci c_ic​i​​ 个代表就餐。
为了使代表们充分交流，希望从同一个单位来的代表不在同一个餐桌就餐。

试设计一个算法，给出满足要求的代表就餐方案。

输入格式

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

输出格式

如果问题有解，在文件第 $1$ 行输出 $1$，否则输出 $0$。
接下来的 $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\le m\le 150,1\le n\le 270$

如果代表团到那个桌子的流量为0 即选了那个桌子

#include <iostream>
#include <cstdio>
#include <cstring>
#include <climits>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;
const int N = 2100;
const int inf = 99999999;
int dis[N], lter[N];
struct node
{
    int to, cap, rev, cap2, id;
};
vector<node>p[N];
void add(int u,int v,int w)
{
    node tmp;
    tmp.to=v, tmp.cap=w,tmp.rev=p[v].size();
    p[u].push_back(tmp);
    tmp.to=u, tmp.cap=0,tmp.rev=p[u].size()-1;
    p[v].push_back(tmp);
    return ;
}
int bfs(int s,int t)
{
    memset(dis,-1,sizeof(dis));
    queue<int>q;
    dis[s]=0;
    q.push(s);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=0; i<p[u].size(); i++)
        {
            int v=p[u][i].to;
            if(dis[v]==-1&&p[u][i].cap>0)
            {
                dis[v]=dis[u]+1;
                q.push(v);
            }
        }
    }
    return dis[t]==-1;
}
int dfs(int s,int t,int flow)
{
    if(s==t) return flow;
    for(int &i=lter[s]; i<p[s].size(); i++)
    {
        int v=p[s][i].to;
        if(dis[v]==dis[s]+1&&p[s][i].cap>0)
        {
            int f=dfs(v,t,min(flow,p[s][i].cap));
            if(f>0)
            {
                p[s][i].cap-=f, p[v][p[s][i].rev].cap+=f;
                return f;
            }
        }
    }
    return 0;
}
int dinic(int s,int e)
{
    int flow=0, f;
    while(1)
    {
        while(bfs(s,e)) return flow;
        memset(lter,0,sizeof(lter));
        while(f=dfs(s,e,inf),f>0)
        {
            flow+=f;
        }
    }
    return flow;
}
int a[N], b[N];
int main()
{
    int n, m;
    scanf("%d %d", &m, &n);
    int s=0, t=m+n+1, sum=0;
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&a[i]);
        add(s,i,a[i]);
        sum+=a[i];
    }
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&b[i]);
        add(m+i,t,b[i]);
    }
    for(int i=1;i<=m;i++)
        for(int j=1;j<=n;j++)   add(i,j+m,1);

    int ans=dinic(s,t);
    if(ans<sum)puts("0");
    else
    {
        puts("1");
        for(int i=1;i<=m;i++)
        {
            for(int j=0;j<p[i].size();j++)
                if(p[i][j].cap==0)    printf("%d ",p[i][j].to-m);
            puts("");
        }
    }
    return 0;
}