P3254 圆桌问题

题意：

假设有来自m 个不同单位的代表参加一次国际会议。每个单位的代表数分别为ri (i =1,2,……,m)。

会议餐厅共有n 张餐桌，每张餐桌可容纳ci (i =1,2,……,n)个代表就餐。

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

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

思路：这是我见过的最友好的网络流了可能，建立源点0和汇点n+m+1，连接每个单位和s，容量是单位人数，每个单位对每个餐桌拆一个点出来限制一下流量，让每个单位去每个餐桌只能有1的流量。然后每个餐桌往终点建立一个边，容量为餐桌的容量。跑dinic就行了，输出也很简单，从每个单位开始，遍历所有的出边，若边的容量变成0，则表示这个边有用，输出。

#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int N = 2e5 + 10;
int n, m, s, t, h[N], cnt, dep[N], cur[N];
struct node {
    int v, w, nt;
} no[N];
void add(int u, int v, int w) {
    no[cnt] = node{v, w, h[u]};
    h[u] = cnt++;
}
int bfs() {
    queue<int> q;
    memset(dep, 0, sizeof dep);
    dep[s] = 1;
    q.push(s);
    while(!q.empty()) {
        int u = q.front();
        q.pop();
        for(int i = h[u]; ~i; i = no[i].nt) {
            int v = no[i].v;
            if(!dep[v] && no[i].w > 0) {
                dep[v] = dep[u] + 1;
                q.push(v);
            }
        }
    }
    return dep[t] > 0;
}
int dfs(int u, int flow) {
    if(u == t)
        return flow;
    for(int &i = cur[u]; ~i; i = no[i].nt) {
        int v = no[i].v;
        if(dep[v] == dep[u] + 1 && no[i].w) {
            int res = dfs(v, min(flow, no[i].w));
            if(res > 0) {
                no[i].w -= res;
                no[i ^ 1].w += res;
                return res;
            }
        }
    }
    return 0;
}
int dinic() {
    int res = 0;
    while(bfs()) {
        for(int i = 0; i <= t; i++)//这必须是t，不然死循环
            cur[i] = h[i];
        while(int d = dfs(s, INF))
            res += d;
    }
    return res;
}
int main() {
    memset(h, -1, sizeof h);
    scanf("%d%d", &m, &n);
    t = n + m + 1;
    int sum = 0;
    for(int k, i = 1; i <= m; i++) {
        scanf("%d", &k);
        sum += k;
        add(s, i, k), add(i, s, 0);
    }
    for(int k, i = 1; i <= n; i++) {
        scanf("%d", &k);
        add(i + m, t, k), add(t, i + m, 0);
    }
    for(int i = 1; i <= m; i++)
        for(int j = m + 1; j <= m + n; j++)
            add(i, j, 1), add(j, i, 0);
    int ans = dinic();
    if(ans == sum) {
        puts("1");
        for(int i = 1; i <= m; i++) {
            for(int j = h[i]; ~j; j = no[j].nt) {
                int v = no[j].v;
                if(v != s && !no[j].w)
                    printf("%d ", v - m);
            }
            puts("");
        }
    } else
        puts("0");
    return 0;
}