网络流24题5. 圆桌问题

圆桌问题

Description

假设有来自 n 个不同单位的代表参加一次国际会议。每个单位的代表数分别为$R_i$。会议餐厅共有 m 张餐桌，每张餐桌可容纳$C_i$个代表就餐。
为了使代表们充分交流，希望从同一个单位来的代表不在同一个餐桌就餐。试设计一个算法，给出满足要求的代表就餐方案。
对于给定的代表数和餐桌数以及餐桌容量，编程计算满足要求的代表就餐方案。

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

题解

同样是因为不能spj。。日后找到方法会更新的。
每个单位为一个点，每个餐桌为一个点。
建图：
从s点向单位点连一条容量为单位人数的边。
从餐桌点向t点连一条容量为餐桌人数的边。
从每个单位点想每个餐桌点连一条容量为1的边。
分析：
在此图中流最大流，必然每张餐桌上只可能至多有一个同一个单位的人。如果最大流等于总人数，那么就可行。
而每条单位连向餐桌的满流边代表该单位有一个人坐在这张桌子上。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

const int N = 450, M = 600000 + 10, inf = 0x3f3f3f3f;

struct Edge{
    int fr, to, cap, flow;
}edg[M];
int hd[N], nxt[M];
int d[N], vis[N], q[N], dfn;
int s, t;
int n, m, ans, tot, sum;
void insert(int u, int v, int w){
    edg[tot].fr = u, edg[tot].to = v, edg[tot].cap = w;
    nxt[tot] = hd[u], hd[u] = tot;
    tot++;
    edg[tot].fr = v, edg[tot].to = u;
    nxt[tot] = hd[v], hd[v] = tot;
    tot++;
}
void init(){
    memset(hd, -1, sizeof(hd));
    scanf("%d %d", &m, &n);
    s = 0, t = m + n + 1;
    int w;
    for(int i = 1; i <= m; i++){
        scanf("%d", &w);
        sum += w;
        insert(s, i, w);
    }
    for(int i = 1; i <= n; i++){
        scanf("%d", &w);
        insert(i + m, t, w);
    }
    for(int i = 1; i <= m; i++)
        for(int j = 1; j <= n; j++)
            insert(i, j + m, 1);
}
bool bfs(){
    int head = 1, tail = 1;
    q[1] = s; vis[s] = ++dfn; d[s] = 0;
    while(head <= tail){
        int u = q[head++];
        for(int i = hd[u]; i >= 0; i = nxt[i]){
            Edge &e = edg[i];
            if(vis[e.to] == dfn || e.cap <= e.flow) continue;
            vis[e.to] = dfn;
            d[e.to] = d[u] + 1;
            q[++tail] = e.to;
        }
    }
    return vis[t] == dfn;
}
int dfs(int x, int a){
    if(x == t || a == 0) return a;
    int flow = 0, f;
    for(int i = hd[x]; i >= 0; i = nxt[i]){
        Edge &e = edg[i];
        if(d[e.to] == d[x] + 1 && (f = dfs(e.to, min(a, e.cap - e.flow))) > 0){
            flow += f;
            e.flow += f;
            edg[i^1].flow -= f;
            a -= f;
            if(a == 0) break;
        }
    }
    return flow;
}
void work(){
    int tmp = 0;
    while(bfs())
        tmp += dfs(s, inf);
    if(tmp == sum) puts("1");
    else puts("0");
}
int main(){
    freopen("prog85.in", "r", stdin);
    freopen("prog85.out", "w", stdout);
    init();
    work();
    return 0;
}