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 54 5 3 53 5 2 6 4

样例输出

11 2 4 51 2 3 4 52 4 51 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;}