P3254 圆桌问题 题解

题目链接

一道模板的带权二分图匹配
工作单位和餐桌分别为该二分图的两组点

• 源点向每个单位所对应的点连一条流量为 $r_i$ 的边，表示有 $r_i$ 个人
• 每个餐桌所对应的点想汇点连一条流量为 $c_i$ 的边，表示可以容纳 $c_i$ 人
• 每一个单位向每一个餐桌连一条流量为 $1$ 的边，表示（这个单位）可以有坐在对应的餐桌。

最后，如果最大匹配小于人数总和，说明没有满足要求的就餐方案，输出 $0$
否则就输出方案

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int Maxn=520,Maxm=100000,inf=0x3f3f3f3f;
int nxt[Maxm],to[Maxm],d[Maxn];
int flow[Maxm],head[Maxn];
int n,m,s,t,ans;
int edgecnt=1,tot;
inline int read()
{
	int s=0,w=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')w=-1;ch=getchar();}
	while(ch>='0' && ch<='9')s=(s<<3)+(s<<1)+(ch^48),ch=getchar();
	return s*w;
}
inline void add(int x,int y,int c)
{
	++edgecnt;
	nxt[edgecnt]=head[x];
	to[edgecnt]=y;
	flow[edgecnt]=c;
	head[x]=edgecnt;
}
bool bfs()
{
	queue <int> q;
	memset(d,0,sizeof(d));
	d[s]=1,q.push(s);
	while(q.size())
	{
		int x=q.front();
		q.pop();
		for(int i=head[x];i;i=nxt[i])
		{
			int y=to[i];
			if(!flow[i] || d[y])continue;
			d[y]=d[x]+1;
			if(y==t)return 1;
			q.push(y);
		}
	}
	return 0;
}
int dinic(int x,int cur)
{
	if(x==t)return cur;
	int ret=cur;
	for(int i=head[x];i && ret;i=nxt[i])
	{
		int y=to[i];
		if(!flow[i] || d[y]!=d[x]+1)continue;
		int tmp=dinic(y,min(ret,flow[i]));
		if(!tmp)d[y]=0;
		flow[i]-=tmp,flow[i^1]+=tmp,ret-=tmp;
	}
	return cur-ret;
}
int main()
{
//	freopen("in.txt","r",stdin);
	n=read(),m=read();
	s=n+m+1,t=s+1;
	for(int i=1;i<=n;++i)
	{
		int c=read();
		tot+=c;
		add(s,i,c);
		add(i,s,0);
		for(int j=n+1;j<=m+n;++j)
		add(i,j,1),add(j,i,0);
	}
	for(int i=n+1;i<=m+n;++i)
	{
		int c=read();
		add(i,t,c);
		add(t,i,0);
	}
	int tmp;
	while(bfs())
	while(tmp=dinic(s,inf))ans+=tmp;
	if(ans!=tot){puts("0");return 0;}
	puts("1");
	for(int x=1;x<=n;++x)
	{
		for(int i=head[x];i;i=nxt[i])
		{
			int y=to[i];
			if(y==s || y==t || flow[i])continue;
			printf("%d ",y-n);
		}
		putchar('\n');
	}
	return 0;
}