题目描述
有来自 mm 个不同单位的代表参加一次国际会议。第 ii 个单位派出了 r_iri 个代表。
会议的餐厅共有 nn 张餐桌，第 ii 张餐桌可容纳 c_ici 个代表就餐。
为了使代表们充分交流，希望从同一个单位来的代表不在同一个餐桌就餐。请给出一个满足要求的代表就餐方案。
输入格式
输入的第一行是用空格隔开的两个整数，分别代表单位的个数 mm 和餐桌的个数 nn。
第二行有 mm 个用空格隔开的整数，第 ii 个整数代表第 ii 个单位的代表人数 r_iri。
第三行有 nn 个用空格隔开的整数，第 ii 个整数代表第 ii 张餐桌能容纳的人数 c_ici。
输出格式
本题存在 Special Judge。
请输出是否存在满足要求的就餐方案，若存在，请给出任意一种可行的方案。
输出的第一行是一个非 00 即 11 的整数，若存在方案则输出 11，否则输出 00。
若存在方案，则对于第 22 到第 (m + 1)(m+1) 行，在第 (i + 1)(i+1) 行输出 r_iri 个整数，代表第 ii 个单位的代表就餐的餐桌编号。
输入输出样例
输入
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
说明/提示
【数据规模与约定】
- 对于 100\%100% 的数据，保证 1 \leq m \leq 1501≤m≤150，1 \leq n \leq 2701≤n≤270，1 \leq r_i, c_i \leq 10^31≤ri,ci≤103。
【提示】
- 请注意输入的第一行先读入 mm 再读入 nn。
思路：
源点向工作单位连边，流量为r[i]，圆桌向汇点连边，流量为c[i]，每个单位向每个圆桌连边，流量为1。dinic跑最大流，如果最大流小于总人数说明没有方案。注意边的输出。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll N = 1e3 + 5;
const ll M = 1e5 + 5; //边集二倍
const ll mod = 1e9 + 7;
const ll inf = 0x3f3f3f3f;
ll head[N], tot, n, m, s, t, r[N];
ll to[N][N], cnt[N];
struct Edge {
ll from, to, next, cap, flow;
}edge[M];
void init() {
tot = 2;
memset(head, -1, sizeof(head));
}
void addedge(ll u, ll v, ll w, ll rw = 0) {
edge[tot].from = u;
edge[tot].to = v;
edge[tot].cap = w;
edge[tot].flow = 0;
edge[tot].next = head[u];
head[u] = tot++;
edge[tot].from = v;
edge[tot].to = u;
edge[tot].cap = rw;
edge[tot].flow = 0;
edge[tot].next = head[v];
head[v] = tot++;
}
ll Q[N];
ll dep[N], cur[N], sta[N]; ///数组cur记录点u之前循环到了哪一条边
bool bfs(ll s, ll t) {
ll fron = 0, tail = 0;
memset(dep, -1, sizeof(dep));
dep[s] = 0;
Q[tail++] = s;
while(fron < tail) {
ll u = Q[fron++];
for(ll i = head[u]; i != -1; i = edge[i].next) {
ll v = edge[i].to;
if(edge[i].cap > edge[i].flow && dep[v] == -1) {
dep[v] = dep[u] + 1;
if(v == t) return true;
Q[tail++] = v;
}
}
}
return false;
}
ll dinic(ll s, ll t) {
ll maxflow = 0;
while(bfs(s, t)) {
for(ll i = 0; i < N; ++i) cur[i] = head[i];
ll u = s, tail = 0;
while(cur[s] != -1) {
if(u == t) {
ll tp = inf;
for(ll i = tail - 1; i >= 0; --i)
tp = min(tp, edge[sta[i]].cap - edge[sta[i]].flow);
maxflow += tp;
for(ll i = tail - 1; i >= 0; --i) {
edge[sta[i]].flow += tp;
edge[sta[i] ^ 1].flow -= tp;
if(edge[sta[i]].cap - edge[sta[i]].flow == 0)
tail = i;
}
u = edge[sta[tail] ^ 1].to;
}
else if(cur[u] != -1 && edge[cur[u]].cap > edge[cur[u]].flow && dep[u] + 1 == dep[edge[cur[u]].to]) {
sta[tail++] = cur[u];
u = edge[cur[u]].to;
}
else {
while(u != s && cur[u] == -1)
u = edge[sta[--tail] ^ 1].to;
cur[u] = edge[cur[u]].next;
}
}
}
return maxflow;
}
int main() {
init();
ll c;
ll sum = 0;
scanf("%lld%lld", &m, &n);
s = 0, t = m + n + 1;
for(ll i = 1; i <= m; ++i) {
scanf("%lld", &r[i]);
addedge(s, i, r[i]);
sum += r[i];
}
for(ll i = 1; i <= n; ++i) {
scanf("%lld", &c);
addedge(i + m, t, c);
}
for(ll i = 1; i <= m; ++i) {
for(ll j = 1; j <= n; ++j) {
addedge(i, j + m, 1);
}
}
ll maxflow = dinic(s, t);
if(maxflow < sum) {
printf("0\n");
return 0;
}
printf("1\n");
memset(to, 0, sizeof(to));
memset(cnt, 0, sizeof(cnt));
for(ll i = 0; i < tot; i += 2) {
ll u = edge[i].from, v = edge[i].to;
if(u == s || v == t) continue;
if(edge[i].flow) {
if(u > m) u -= m;
if(v > m) v -= m;
to[u][++cnt[u]] = v;
}
}
for(ll i = 1; i <= m; ++i) {
for(ll j = 1; j <= r[i]; ++j) {
if(j > 1) printf(" ");
printf("%lld", to[i][j]);
}
printf("\n");
}
return 0;
}