网络流二十四题之七 —— 试题库问题（DATA）

试题库问题

---

Description

假设一个试题库中有 $n$ 道试题。
每道试题都标明了所属类别。
同一道题可能有多个类别属性。
现要从题库中抽取 $m$ 道题组成试卷，并要求试卷包含指定类型的试题。
试设计一个满足要求的组卷算法。

---

Input

输入文件第 $1$ 行有 $2$ 个正整数 $n$ 和 $k$ $(2 <=k<= 20, k<=n<= 1000)$。
$k$ 表示题库中试题类型总数，$n$ 表示题库中试题总数。
第 $2$ 行有 $k$ 个正整数，第 $i$ 个正整数表示要选出的类型 $i$ 的题数。
这 $k$ 个数相加就是要选出的总题数 $m$。
接下来的 $n$行给出了题库中每个试题的类型信息。
每行的第 $1$ 个正整数 $p$ 表明该题可以属于 $p$ 类，接着的 $p$ 个数是该题所属的类型号。

---

Output

输出文件第 $i$ 行输出 $“i：”$ 后接类型 $i$的题号。
如果有多个满足要求的方案，只要输出 $1$ 个方案。
如果问题无解，则输出 $“No\ Solution!”$。

---

Sample Input

3 15
3 3 4
2 1 2
1 3
1 3
1 3
1 3
3 1 2 3
2 2 3
2 1 3
1 2
1 2
2 1 2
2 1 3
2 1 2
1 1
3 1 2 3

---

Sample Output

1: 1 6 8
2: 7 9 10
3: 2 3 4 5

---

Solution

很简单的网络流题目。
建一个图，然后跑最大流就可以了。
代码写得很丑，不要介意。

---

Code

[cpp]

1. #include <iostream>  
2. #include <cstdio>  
3. #include <cstring>  
4. #include <queue>  
5.   
6. #define ANOTHER 2000   
7. #define st 0  
8. #define tt 5000  
9. #define INF 0x3f3f3f3f  
10.   
11. #define Min(x,y) ((x)<(y)?(x):(y))  
12. #define Max(x,y) ((x)>(y)?(x):(y))  
13.   
14. using namespace std;  
15.   
16. int n,k,m,cnt;  
17. int s[1010];  
18.   
19. int head[10010],nxt[100010],data[100010],flow[100010];  
20. int dis[10010];  
21. queue<int>q;  
22.   
23. void add(int x,int y,int z){  
24.     nxt[cnt]=head[x];data[cnt]=y;flow[cnt]=z;head[x]=cnt++;  
25.     nxt[cnt]=head[y];data[cnt]=x;flow[cnt]=0;head[y]=cnt++;   
26. }  
27.   
28. bool BFS(){  
29.     memset(dis,-1,sizeof dis);  
30.     q.push(st);  
31.     dis[st]=0;  
32.     while(!q.empty()){  
33.         int now=q.front();  
34.         q.pop();  
35.         for(int i=head[now];i!=-1;i=nxt[i])  
36.             if(flow[i]&&dis[data[i]]==-1){  
37.                 dis[data[i]]=dis[now]+1;  
38.                 q.push(data[i]);   
39.             }  
40.     }  
41.     return dis[tt]>0;  
42. }  
43.   
44. int dfs(int now,int low){  
45.     if(now==tt)return low;  
46.     int Flow;  
47.     for(int i=head[now];i!=-1;i=nxt[i]){  
48.         if(flow[i]&&dis[data[i]]==dis[now]+1){  
49.             if(Flow=dfs(data[i],Min(low,flow[i]))){  
50.                 flow[i]-=Flow;  
51.                 flow[i^1]+=Flow;  
52.                 return Flow;   
53.             }  
54.         }  
55.     }  
56.     return 0;  
57. }  
58.   
59. int main(){  
60.     memset(head,-1,sizeof head);  
61.     scanf(”%d%d”,&k,&n);  
62.     for(int i=1;i<=k;i++){  
63.         scanf(”%d”,&s[i]);  
64.         add(i+ANOTHER+ANOTHER,i+ANOTHER,s[i]);  
65.         add(i+ANOTHER,tt,INF);  
66.         m+=s[i];  
67.     }  
68.     for(int i=1;i<=n;i++){  
69.         int p;  
70.         add(st,i,INF);  
71.         scanf(”%d”,&p);  
72.         for(int j=1;j<=p;j++){  
73.             int x;  
74.             scanf(”%d”,&x);  
75.             add(i,x+ANOTHER+ANOTHER,1);  
76.         }  
77.     }  
78.     int weight=0;  
79.     while(BFS()){  
80.         int flag;  
81.         while(flag=dfs(st,INF))weight+=flag;  
82.     }  
83.     if(weight!=m)printf(“No Solution!”);  
84.     else{  
85.         for(int i=1;i<=k;i++){  
86.             printf(”%d: ”,i);  
87.             for(int j=1;j<=n;j++){  
88.                 for(int l=head[j];l!=-1;l=nxt[l]){  
89.                     if(data[l]==i+ANOTHER+ANOTHER&&!flow[l]){  
90.                         printf(”%d ”,j);  
91.                         break;  
92.                     }  
93.                 }  
94.             }  
95.             printf(”\n”);  
96.         }  
97.     }  
98.     return 0;  
99. }