[网络流24题] 洛谷P3254 圆桌问题 最大流

给出$m\le 150$个单位，$n\le 270$个餐桌，并且给出了每个单位的人数以及每个餐桌容纳的人数。求一个满足的方案，不能的话就直接输出$0$。
这个题简单了。就是直接源点流人数的流量到单位，单位流$1$的流量到餐桌，餐桌流餐桌的人数到汇点。判断是不是人数能流完，流完说明有解。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
int sum=0;
namespace Dinic {
	const int N=1e4+3,M=1e5+3; 
	int front[N],head[N],dep[N],tot=-1,n,S,T;
	int m;
	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);
		}
		vector<int> g[N];
		if(ans<sum) puts("0");
		else {
			puts("1");
			n=n-m-2;
			for(int i=1;i<=m;i++) {
				for(int j=front[i];j!=-1;j=e[j].nxt) {
					if(e[j].v==S) continue;
					else if(e[j].w==0) g[i].push_back(e[j].v-m);
				}
			}
			for(int i=1;i<=m;i++) {
				for(int j=0;j<g[i].size();j++) {
					printf("%d%c",g[i][j],j+1==g[i].size()?'\n':' ');
				}
			}
		} 
		
	}
}
int r[155],c[275];  
int main() {
	int m,n;
	scanf("%d%d",&m,&n);
	for(int i=1;i<=m;i++) {
		scanf("%d",&r[i]);
		sum+=r[i];
	} 
	for(int i=1;i<=n;i++)
		scanf("%d",&c[i]);
	int S=m+n+1;
	int T=m+n+2;
	Dinic::S=S;
	Dinic::T=T;
	Dinic::n=n+m+2; 
	Dinic::m=m;
	Dinic::init();
	for(int i=1;i<=m;i++)
		Dinic::add(S,i,r[i]);
	for(int i=m+1;i<=m+n;i++)
		Dinic::add(i,T,c[i-m]);
	for(int i=1;i<=m;i++) {
		for(int j=m+1;j<=m+n;j++) {
			Dinic::add(i,j,1);
		}
	}
	Dinic::solve();
	return 0;
}