[网络流24题] 洛谷P2763 试题库问题 最大流

给出$n\le 1e3$个问题，然后每个问题有若干个类型，共有$k\le 20$种类型，然后要组$m$个题，给出每个类型的题数$t_i$，${\sum }_{i=1}^k{t}_i=m$。求一种满足条件的方案。
源点向所有的题连$1$的流量，题向它的类型也连$1$的流量，它的每种类型向汇点连所需类型个数的流量。如果满流则说明方案有解。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
int sum=0;
vector<int> g[21]; 
namespace Dinic {
	const int N=2e3+3,M=1e5+3; 
	int front[N],head[N],dep[N],tot=-1,n,m,S,T;
	struct edge{ int v,nxt;ll w; }e[2*M];
	void init() {
		tot=-1;
		for(int i=1;i<=n;i++) front[i]=-1;
	}
	void add(int u,int v,ll w) {
		e[++tot].v=v,e[tot].w=w,e[tot].nxt=front[u],front[u]=tot;
		e[++tot].v=u,e[tot].w=0,e[tot].nxt=front[v],front[v]=tot;
	}
	bool bfs() {
		for(int i=1;i<=n;i++) dep[i]=0;
		queue<int> q;
		q.push(S);
		dep[S]=1;
		while(!q.empty()) {
			int u=q.front();
			q.pop();
			for(int i=front[u];i!=-1;i=e[i].nxt) {
				int v=e[i].v;
				if(dep[v]==0&&e[i].w) {
					dep[v]=dep[u]+1;
					q.push(v); 
				}
			}
		}
		return dep[T];
	}
	ll dfs(int cur,ll dis=1e18) {
		if(cur==T) return dis;
		ll flow=0,sum=0;
		for(int &i=head[cur];i!=-1;i=e[i].nxt) {
			int v=e[i].v;
			if(dep[v]==dep[cur]+1&&e[i].w) {
				flow=dfs(v,min(dis,e[i].w));
				dis-=flow,sum+=flow;
				e[i].w-=flow,e[i^1].w+=flow;
				if(!dis) break;
			}
		}
		return sum;
	}
	void solve() {
		ll ans=0;
		while(bfs()) {
			for(int i=1;i<=n;i++) head[i]=front[i];
			ans+=dfs(S);
		}
		if(ans!=sum) {
			printf("No Solution!");
		} 
		else {
			for(int i=1;i<=n-m-2;i++) {
				for(int j=front[i];j!=-1;j=e[j].nxt) {
					if(e[j].v==S) continue;
					if(e[j].w==0) {
						g[e[j].v-(n-m-2)].push_back(i);
					}
				} 
			}
			for(int i=1;i<=m;i++) {
				printf("%d: ",i);
				for(int j=0;j<g[i].size();j++) {
					printf("%d%c",g[i][j],j+1==g[i].size()?'\n':' '); 
				}
			}
		}
	}
}
int rec[22];
int main() {
	int m,n; 
	scanf("%d%d",&m,&n);
	for(int i=1;i<=m;i++) {
		scanf("%d",&rec[i]);
		sum+=rec[i];
	}
	Dinic::n=n+m+2;
	Dinic::m=m;
	Dinic::init();
	int S=n+m+1;
	int T=n+m+2;
	Dinic::S=S;
	Dinic::T=T;
	for(int tot,u=1;u<=n;u++) {
		scanf("%d",&tot);
		for(int v,j=1;j<=tot;j++) {
			scanf("%d",&v);v+=n;
			Dinic::add(u,v,1);
		}
	}
	for(int i=1;i<=n;i++) 
		Dinic::add(S,i,1);
	for(int i=n+1;i<=n+m;i++) 
		Dinic::add(i,T,rec[i-n]);
	Dinic::solve();
	return 0;
}