题意:
假设一个试题库中有n道试题。每道试题都标明了所属类别。同一道题可能有多个类别属性。现要从题库中抽取m 道题组成试卷。并要求试卷包含指定类型的试题。试设计一个满足要求的组卷算法。
«编程任务:
对于给定的组卷要求,计算满足要求的组卷方案。
思路:
建图,源点向每个类型建边,容量就是这个类型所需要的试题数,然后对每个关系建立一条边,然后把每个题目和终点再连一条边,容量为1,跑出来的最大流就是答案。
#include<bits/stdc++.h>
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int N = 2e5 + 10;
int n, m, s, t, h[N], cnt, dep[N], cur[N], a[N], dp[N];
struct node {
int v, w, nt;
} no[N];
void add(int u, int v, int w) {
no[cnt] = node{v, w, h[u]};
h[u] = cnt++;
no[cnt] = node{u, 0, h[v]};
h[v] = cnt++;
}
int bfs() {
queue<int> q;
memset(dep, 0, sizeof dep);
dep[s] = 1;
q.push(s);
while(!q.empty()) {
int u = q.front();
q.pop();
for(int i = h[u]; ~i; i = no[i].nt) {
int v = no[i].v;
if(!dep[v] && no[i].w > 0) {
dep[v] = dep[u] + 1;
q.push(v);
}
}
}
return dep[t] > 0;
}
int dfs(int u, int flow) {
if(u == t)
return flow;
for(int &i = cur[u]; ~i; i = no[i].nt) {
int v = no[i].v;
if(dep[v] == dep[u] + 1 && no[i].w) {
int res = dfs(v, min(flow, no[i].w));
if(res > 0) {
no[i].w -= res;
no[i ^ 1].w += res;
return res;
}
}
}
return 0;
}
int dinic() {
int res = 0;
while(bfs()) {
for(int i = 0; i <= t; i++)//这必须是t,不然死循环
cur[i] = h[i];
while(int d = dfs(s, INF))
res += d;
}
return res;
}
int main() {
int k, ti = 0;
memset(h, -1, sizeof h);
scanf("%d%d", &k, &n);
for(int y, i = 1; i <= k; i++) {
scanf("%d", &y);
ti += y;
add(s, i, y);
}
t = n + k + 1;
for(int ty, i = 1; i <= n; i++) {
scanf("%d", &ty);
add(i + k, t, 1);
for(int lei, j = 1; j <= ty; j++) {
scanf("%d", &lei);
add(lei, i + k, 1);
}
}
int ans = dinic();
if(ans == ti) {
for(int i = 1; i <= k; i++) {
printf("%d: ", i);
for(int j = h[i]; ~j; j = no[j].nt) {
int v = no[j].v;
if(v != s && !no[j].w)
printf("%d ", v - k);
}
puts("");
}
} else
puts("No Solution!");
return 0;
}