[洛谷P2763] 试题库问题（最大流）

https://www.luogu.org/problemnew/show/P2763

典型最大流问题。建立源点，向每个题连容量为1的边；从每道题向其所属分类连容量为1的边；建立汇点，从每个分类向其连容量为该分类所要求的题数的边。跑最大流看是否等于总题数，若小于则无解。有解的话从源点搜索一边，容量为0的边就是用过的，记录一下用过的题和其属于的分类，然后按顺序输出就好了。输出时即使一个分类下题数为0，也还是要把这个分类输出来。

#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int N=1100,M=44000,K=30,inf=0x7fffffff;
struct edge{
	int y,f,next;
}data[M];
int k,n,m,s,t,num,h[N],dep[N],cur[N];
bool vis[N],ans[K][N];
queue<int> q;
inline int read(){
	int x=0,f=0;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return f?-x:x;
}
inline void add(int x,int y,int f){
	data[++num].y=y;data[num].f=f;data[num].next=h[x];h[x]=num;
	data[++num].y=x;data[num].f=0;data[num].next=h[y];h[y]=num;
}
bool bfs(int s,int t){
	memset(dep,0,sizeof dep);
	for(int i=s;i<=t;i++)cur[i]=h[i];
	while(!q.empty())q.pop();
	q.push(s);dep[s]=1;
	while(!q.empty()){
		int u=q.front();q.pop();
		for(int i=h[u];i!=-1;i=data[i].next){
			int v=data[i].y;
			if(!dep[v]&&data[i].f>0){
				dep[v]=dep[u]+1;
				q.push(v);
			}
		}
	}
	return dep[t];
}
int dfs(int u,int t,int lim){
	if(u==t)return lim;
	int res=0;
	for(int& i=cur[u];i!=-1;i=data[i].next){
		int v=data[i].y;
		if(dep[v]==dep[u]+1&&data[i].f>0){
			int fmax=dfs(v,t,min(lim-res,data[i].f));
			if(fmax>0){
				data[i].f-=fmax;data[i^1].f+=fmax;
				res+=fmax;
				if(res==lim)return res;
			}
		}
	}
	return res;
}
int dinic(int s,int t){
	int mf=0;
	while(bfs(s,t))mf+=dfs(s,t,inf);
	return mf;
}
int main(){
	k=read();n=read();s=0;t=k+n+1;
	memset(h,-1,sizeof h);num=-1;
	for(int i=1;i<=n;i++)add(s,i,1);
	for(int x,i=1;i<=k;i++){x=read();m+=x;add(i+n,t,x);}
	for(int p,i=1;i<=n;i++){
		p=read();
		for(int x,j=1;j<=p;j++){x=read();add(i,x+n,1);}
	}
	if(dinic(s,t)<m)printf("No Solution!");
	else{
		memset(ans,false,sizeof ans);
		for(int i=h[s];i!=-1;i=data[i].next)
			if(!data[i].f)
				for(int j=h[data[i].y];j!=-1;j=data[j].next)
					if(!data[j].f){ans[data[j].y-n][data[i].y]=true;break;}
		for(int i=1;i<=k;i++){
			printf("%d:",i);
			for(int j=1;j<=n;j++)if(ans[i][j])printf(" %d",j);
			printf("\n");
		}
	}
	return 0;
}