题目描述
假设有来自 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;
}