LOJ6004圆桌聚餐（网络流24题）

题目描述
假设有来自 n 个不同单位的代表参加一次国际会议。每个单位的代表数分别为 ri
​​ 。会议餐厅共有 m 张餐桌，每张餐桌可容纳 ci 个代表就餐。为了使代表们充分交流，希望从同一个单位来的代表不在同一个餐桌就餐。
试设计一个算法，给出满足要求的代表就餐方案。

输入格式
文件第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≤m≤150,1≤n≤270

---

原点向单位连人数的边，单位向所有餐桌连1的边，餐桌向汇点连容量的边
跑一遍最大流判一下是否满流即可

#include<bits/stdc++.h>
using namespace std;
#define ll long long
ll ans , sum; 
int n , m , S , T;
int linkk[11000] , t;
int dep[11000];
bool flag;
const ll oo = 1000000000000000000LL;
struct node{
    int n , y , v;
}e[100100];
int read()
{
    int sum = 0;char c = getchar();bool flag = true;
    while( c < '0' || c > '9' ) {if(c == '-') flag = false;c = getchar();}
    while( c >= '0' && c <= '9' ) sum = sum * 10 + c - 48 , c = getchar();
    if(flag)  return sum;
     else return -sum;
}  
void insert(int x,int y,int z)
{
    e[++t].y = y;e[t].n = linkk[x];e[t].v = z;linkk[x] = t;
    e[++t].y = x;e[t].n = linkk[y];e[t].v = 0;linkk[y] = t;
}
void init()
{
    t = 1;
    n = read();m = read();//1~n单位 n+1~n+m餐桌 
    S = n + m + 1;T = n + m + 2;
    for(int i = 1;i <= n;++i)
    {
        int x = read();sum += x;
        insert(S , i , x);
    }
    for(int i = 1;i <= m;++i)
    {
        int x = read();
        insert(i + n , T , x);
    }
    for(int i = 1;i <= n;++i)
        for(int j = 1;j <= m;++j)
            insert(i , j + n , 1);
    return;
}
ll dfs(int x,ll lazy)
{
    if(x == T) return lazy;
    ll nowlazy = 0;
    ll d = 0;
    for(int i = linkk[x];i&&nowlazy < lazy;i = e[i].n)
    {
        int y = e[i].y;
        if(dep[y] == dep[x] + 1 && e[i].v > 0)
            if(d = dfs(y,min(1ll*e[i].v , lazy - nowlazy)))
            {
                nowlazy += d;
                e[i].v -= d;
                e[i^1].v += d; 
            }
    }
    if(nowlazy == 0) dep[x] = -1;
    return nowlazy;
}
queue<int>q;
bool bfs()
{
    memset(dep,-1,sizeof(dep));
    dep[S] = 0;q.push(S);
    while(q.size())
    {
        int v = q.front();
        for(int i = linkk[v];i;i = e[i].n)
            if(dep[e[i].y] == -1 && e[i].v > 0) 
                dep[e[i].y] = dep[v] + 1,q.push(e[i].y);
        q.pop();
    }
    if(dep[T] == -1) return false;
    return true;
}
int main()
{
    init();
    while(bfs())
    {       
        ll d;
        while(d = dfs(S,oo)) ans += d;
    }
    if(ans == sum) printf("1\n");
        else 
        {
            printf("0\n");
            return 0;       
        }
    for(int i = 1;i <= n;++i)
    {
        for(int j = linkk[i];j;j = e[j].n)
            if(e[j].v == 0)
                printf("%d ",e[j].y - n);
        printf("\n");
    }
    return 0;
}