kyeremal-网络流24题T5-圆桌问题

原题：

圆桌问题
问题描述: 假设有来自 n 个不同单位的代表参加一次国际会议。每个单位的代表数分别为ri ,i =1,2,...,n。会议餐厅共有 m 张餐桌,每张餐桌可容纳ci (i =1,2,...,m)个代表就餐。为了使代表们充分交流,希望从同一个单位来的代表不在同一个餐桌就餐。试设计一个算法, 给出满足要求的代表就餐方案。
编程任务: 对于给定的代表数和餐桌数以及餐桌容量,编程计算满足要求的代表就餐方案。
́数据输入: 由文件 input.txt 提供输入数据。文件第 1 行有 2 个正整数 m 和 n,m 表示单位数,n 表示餐桌数,1<=m
结果输出:
程序运行结束时,将代表就餐方案输出到文件 output.txt 中。如果问题有解,在文件第 1 行输出 1,否则输出 0。接下来的 m 行给出每个单位代表的就餐桌号。如果有多个满足要求的方案,只要输出 1 个方案。
输入文件示例
input.txt
4 5
4 5 3 5
3 5 2 6 4
输出文件示例
output.txt
1
1 2 4 5
1 2 3 4 5
2 4 5
1 2 3 4 5

分析：

算法1：每个单位建立一个节点，每个圆桌也建立一个节点，对于每个单位i与圆桌j，则i->j连接一条边，那么问题便转化为二分图多重匹配问题，建立超级源S与超级汇T，S向每个单位i连接一条流量为ri的边，每一个圆桌j向T连接一条流量为ci的边，单位i与圆桌j间边的流量为1.做一遍最大流即可。

算法2：考虑到要使所有人合法坐入，枚举每个单位的每一个人，坐在可以坐的剩余座位最多的圆桌，易证正确性。

code：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <cstring>
#include <algorithm>
#include <queue>
#include <vector>

using namespace std;

#define rep(i, l, r) for (int i = l; i <= r; i++)
#define REP(i, l, r) for (int i = l; i >= r; i--)
#define MAXN 1010

int n, m, N = -1, sumA = 0, first[MAXN], next[MAXN], S, T, dis[MAXN];
struct tlist {int x, y, f;} a[MAXN];
queue<int> q;
vector<int> v;

inline int min(int a, int b) {return a<b ? a : b;}

inline void add(int x, int y, int f) {
	a[++N].x = x, a[N].y = y, a[N].f = f, next[N] = first[x], first[x] = N;
	a[++N].x = y, a[N].y = x, a[N].f = 0, next[N] = first[y], first[y] = N;
}

inline bool bfs() {
	while (!q.empty()) q.pop();
	memset(dis, -1, sizeof(dis));
	q.push(S);
	dis[S] = 0;
	while (!q.empty()) {
		int x = q.front();
		q.pop();
		for (int i = first[x]; ~i; i = next[i])
			if (!~dis[a[i].y] && a[i].f) {
				dis[a[i].y] = dis[x] + 1;
				q.push(a[i].y);
			}
	}
	return ~dis[T];
}

inline int find(int x, int low) {
	if (x == T) return low;
	int temp, sum = 0;
	for (int i = first[x]; ~i; i = next[i])
		if ((dis[a[i].y] == dis[x] + 1) && (a[i].f) && (temp = find(a[i].y, min(low-sum, a[i].f)))) {
			a[i].f -= temp;
			a[i^1].f += temp;
			sum += temp;
		}
	return sum;
}

inline int dinic(int begin, int end) {
	int ans = 0, temp;
	S = begin, T = end;
	while (bfs())
		while (temp = find(S, 0x7fffffff))
			ans += temp;
	return ans;
}

inline void writeln(int x) {
	v.clear();
	for (int i = first[x]; ~i; i = next[i])
		if ((a[i].y <= n+m) && (a[i].y >= n+1) && (!a[i].f))
			v.push_back(a[i].y - n);
	sort(v.begin(), v.end());
	rep(i, 0, int(v.size())-1) printf("%d", v[i]), cout << ((i == int(v.size())-1) ? "\n" : " ");
}

int main() {
	memset(first, -1, sizeof(first));
	memset(next, -1, sizeof(next));
	int temp;
	cin >> n >> m;
	rep(i, 1, n) scanf("%d", &temp), sumA += temp, add(0, i, temp);
	rep(i, 1, m) scanf("%d", &temp), add(n+i, n+m+1, temp);
	rep(i, 1, n)
		rep(j, 1, m)
			add(i, n+j, 1);
	if (dinic(0, n+m+1)-sumA >= 0) {
		puts("1");
		rep(i, 1, n) writeln(i);
	}
	else puts("0");

	return 0;
}