网络流 - 二分图的多重匹配问题 - 圆桌问题 - 网络流24题 - 洛谷 P3254

网络流 - 二分图的多重匹配问题 - 圆桌问题 - 网络流24题 - 洛谷 P3254

题目描述

$有来自m个不同单位的代表参加一次国际会议。第i个单位派出了r_i个代表。$

$会议的餐厅共有n张餐桌，第i张餐桌可容纳c_i个代表就餐。$

$为了使代表们充分交流，希望从同一个单位来的代表不在同一个餐桌就餐。请给出一个满足要求的代表就餐方案。$

输入格式

$输入的第一行是用空格隔开的两个整数，分别代表单位的个数m和餐桌的个数n。$

$第二行有m个用空格隔开的整数，第i个整数代表第i个单位的代表人数r_i。$

$第三行有n个用空格隔开的整数，第i个整数代表第i张餐桌能容纳的人数c_i。$

输出格式

本题存在 Special Judge。

$请输出是否存在满足要求的就餐方案，若存在，请给出任意一种可行的方案。$

$输出的第一行是一个非0即1的整数，若存在方案则输出1，否则输出0。$

$若存在方案，则对于第2到第(m+1)行，在第(i+1)行输出r_i个整数，代表第i个单位的代表就餐的餐桌编号。$

输入输出样例

输入 #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

说明/提示

【数据规模与约定】

$对于100\%的数据，保证1\le m\le 150，1\le n\le 270，1\le r_i,c_i\le 10^3。$

分析：

$二分图的多重匹配问题。$

$即同一点可以与不同的点匹配。$

$题意即：两组点之间进行匹配，每个点都有匹配次数的上限值，求最大匹配。$

$从源点S向一侧的点（代表）连接权值为r_i的正向边，权值为0的反向边。$

$从另一侧点（餐桌）向汇点T连接一条权值为c_i的正向边，权值为0的反向边。$

$从代表中的所有点向餐桌中的所有点连接一条权值为1的正向边，权值为0的反向边，这确保了同一个点只能与另一个点匹配一次。$

$然后在建立好的残留网络中跑最大流。$

$若最大流与人数总和相等，则说明所有的人都能安排入座。$

$然后对每个点（代表），遍历其相邻的边，若某条边满流，则输出这条边的端点编号。$

$注意，编号要减去偏移量（这取决于建图的时候的顺序）。$

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

const int N = 150 + 270 + 10, M = (150 * 270 + 150 + 270 + 10) * 2, inf = 1e9;

int n, m, S, T;
int e[M], ne[M], f[M], h[N], idx;
int q[N], cur[N], d[N];

void add(int a, int b, int c)
{
    e[idx] = b, f[idx] = c, ne[idx] = h[a], h[a] = idx ++;
}

bool bfs()
{
    int hh = 0, tt = -1;
    memset(d, -1, sizeof d);
    d[S] = 0, q[++ tt] = S, cur[S] = h[S];
    
    while(hh <= tt)
    {
        int u = q[hh ++];
        
        for(int i = h[u]; ~i; i = ne[i])
        {
            int v = e[i];
            
            if(d[v] == -1 && f[i] > 0)
            {
                d[v] = d[u] + 1;
                cur[v] = h[v];
                if(v == T) return true;
                q[++ tt] = v;
            }
        }
    }
    return false;
}

int find(int u, int limit)
{
    if(u == T) return limit;
    
    int flow = 0, t;
    for(int i = cur[u]; ~i && flow < limit; i = ne[i])
    {
        int v = e[i];
        cur[u] = i;
        
        if(d[v] == d[u] + 1 && f[i] > 0)
        {
            t = find(v, min(f[i], limit - flow));
            if(!t) d[v] = -1;
            f[i] -= t, f[i ^ 1] += t, flow += t;
        }
    }
    return flow;
}

int dinic()
{
    int res = 0, flow;
    while(bfs()) while(flow = find(S, inf)) res += flow;
    return res;
}

int main()
{
    scanf("%d%d", &n, &m);
    S = 0, T = 150 + 270 + 1;
    memset(h, -1, sizeof h);
    int sum = 0;
    for(int i = 1; i <= n; i ++)
    {
        int r;
        scanf("%d", &r);
        sum += r;
        add(S, i, r), add(i, S, 0);
    }
    for(int i = n + 1; i <= n + m; i ++)
    {
        int c;
        scanf("%d", &c);
        add(i, T, c), add(T, i, 0);
    }
    for(int i = 1; i <= n; i ++)
        for(int j = n + 1; j <= n + m; j ++)
            add(i, j, 1), add(j, i, 0);
    
    int res = dinic();
    
    if(res == sum)
    {
        puts("1");
        for(int i = 1; i <= n; i ++)
        {
            for(int j = h[i]; ~j; j = ne[j])
                if(!f[j])
                    printf("%d ", e[j] - n);
            puts("");
        }
    }
    else puts("0");

    return 0;
}