COGS 732. [网络流24题] 试题库

«问题描述：
假设一个试题库中有n道试题。每道试题都标明了所属类别。同一道题可能有多个类别属性。现要从题库中抽取m 道题组成试卷。并要求试卷包含指定类型的试题。试设计一个满足要求的组卷算法。
«编程任务：
对于给定的组卷要求，计算满足要求的组卷方案。
«数据输入：
由文件testlib.in提供输入数据。文件第1行有2个正整数k和n (2 <=k<= 20, k<=n<= 1000)k 表示题库中试题类型总数，n 表示题库中试题总数。第2 行有k 个正整数，第i 个正整数表示要选出的类型i 的题数。这k个数相加就是要选出的总题数m。接下来的n行给出了题库中每个试题的类型信息。每行的第1 个正整数p表明该题可以属于p类，接着的p个数是该题所属的类型号。
«结果输出:
程序运行结束时，将组卷方案输出到文件testlib.out 中。文件第i 行输出 “i：”后接类型i的题号。如果有多个满足要求的方案，只要输出1 个方案。如果问题无解，则输出“NoSolution!”。
输入文件示例
testlib.in
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

输出文件示例

testlib.out

1: 1 6 8
2: 7 9 10

3: 2 3 4 5

这题似乎与圆桌聚餐一样，就不细说了。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
using namespace std;
const int N=1025;
const int inf=1e9+7;
int m,n,s,t,cnt=1,sum,hd[N],d[N];
queue<int>q;
struct edge
{
	int to,nxt,f;
}v[2*N*N];
void addedge(int x,int y,int z)
{
	v[++cnt].to=y;
	v[cnt].f=z;
	v[cnt].nxt=hd[x];
	hd[x]=cnt;
}
void addedges(int x,int y,int z)
{
	addedge(x,y,z),addedge(y,x,0);
}
bool bfs()
{
	memset(d,0,sizeof(d));
	d[s]=1;
	q.push(s);
	while(!q.empty())
	{
		int u=q.front();
		q.pop();
		for(int i=hd[u];i;i=v[i].nxt)
			if(v[i].f&&!d[v[i].to])
			{
				d[v[i].to]=d[u]+1;
				q.push(v[i].to);
			}
	}
	return d[t];
}
int dfs(int u,int lft)
{
	if(u==t||lft==0)
		return lft;
	int r=lft;
	for(int i=hd[u];i;i=v[i].nxt)
		if(v[i].f&&r&&d[v[i].to]==d[u]+1)
		{
			int w=dfs(v[i].to,min(r,v[i].f));
			v[i].f-=w,v[i^1].f+=w,r-=w;
			if(!r)
				return lft;
		}
	if(r==lft)
		d[u]=0;
	return lft-r;
}
int main()
{
	freopen("testlib.in","r",stdin);
	freopen("testlib.out","w",stdout);
	scanf("%d%d",&m,&n);
	s=0,t=n+m+1;
	for(int i=1;i<=m;i++)
	{
		int x;
		scanf("%d",&x);
		sum+=x;
		addedges(s,i,x);
	}
	for(int i=1;i<=n;i++)
	{
		int x,y;
		scanf("%d",&x);
		while(x--)
		{
			scanf("%d",&y);
			addedges(y,i+m,1);
		}
		addedges(i+m,t,1);
	}
	while(bfs())
		sum-=dfs(s,inf);
	if(sum>0)
		printf("NoSolution!\n");
	else
	{
		for(int i=1;i<=m;i++)
		{
			printf("%d: ",i);
			for(int j=hd[i];j;j=v[j].nxt)
				if(!v[j].f)
					printf("%d ",v[j].to-m);
			printf("\n");
		}
	}
	return 0;
}