（网络流）洛谷P3254圆桌问题

洛谷P3254圆桌问题

思路：

最近做到的最舒服的网络流了。
源点向每个单位建边，权值为$r_i$;
每个单位向每个餐桌建边，权值为$1$;
每个餐桌向汇点建边，权值为$c_i$;
如果跑最大流的结果为$\sum r_i$的话就是可以分配的。

代码：

#include<bits/stdc++.h>
#define pii pair<int,int>
#define int long long
#define cl(x,y) memset(x,y,sizeof(x))
#define ct cerr<<"Time elapsed:"<<1.0*clock()/CLOCKS_PER_SEC<<"s.\n";
const int N=5e5+210;
const int mod=1e9+7;
const int inf=0x3f3f3f3f;
const int minn=0xc0c0c0c0;
using namespace std;
struct edge
{
	int u,v,w;
}maze[N<<1];
int len=1,head[N]={0};
int dep[N];//深度 
void add(int u,int v,int w)
{
	maze[++len]={head[u],v,w};
	head[u]=len;
}
void inc(int u,int v,int w)
{
	add(u,v,w);
	add(v,u,0);
} 
int dfs(int u,int f,int t)
{
	int ans=0,i;
	if(u==t)
		return f;
	for(i=head[u];i && f;i=maze[i].u)
	{
		int v=maze[i].v,w=maze[i].w;
		if(dep[v]==dep[u]+1 && w)//符合深度关系且能流 
		{
			int sum=dfs(v,min(f,w),t);
			maze[i].w-=sum;
			maze[i^1].w+=sum;
			f-=sum;
			ans+=sum;
		}	
	}
	if(!ans)
		dep[u]=-2;
	return ans;
}
int bfs(int s,int t)
{
	queue<int> q;
	cl(dep,0);
	dep[s]=1;//源点深度为1
	q.push(s);
	while(!q.empty())
	{
		int u=q.front(),i;
		q.pop();
		for(i=head[u];i;i=maze[i].u)
		{
			int v=maze[i].v,w=maze[i].w;
			if(w && !dep[v])//有深度且能流 
			{
				dep[v]=dep[u]+1;
				q.push(v); 
			}
		}
	}
	return dep[t];
}
int dinic(int s,int t)
{
	int ans=0;
	while(bfs(s,t))
		ans+=dfs(s,inf,t);
	return ans;
}
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);cout.tie(0);
	int n,m,i,j,sum=0;
	cin>>m>>n;
	int s=0,t=n+m+1;
	for(i=1;i<=m;i++)
	{
		int w;
		cin>>w;
		sum+=w;
		inc(s,i,w);
	}
	for(i=1;i<=n;i++)
	{
		int w;
		cin>>w;	
		inc(m+i,t,w);
	}
	for(i=1;i<=m;i++)
		for(j=1;j<=n;j++)
			inc(i,j+m,1);
	if(dinic(s,t)==sum)
	{
		cout<<1<<endl;
		for(i=1;i<=m;i++)
		{
			for(j=head[i];j;j=maze[j].u)
				if(maze[j].v!=s && maze[j].w==0)
					cout<<maze[j].v-m<<" ";
			cout<<endl;
		}
	}
	else
		cout<<0<<endl;
	return 0;
}