[Loj]#6004. 「網絡流 24 題」圓桌聚餐

建立源點s與所有單位連邊，建立匯點T將所有餐桌與t相連，再將單位與餐桌各連上容量爲1的邊。

跑一遍最大流即可。

輸出路徑時，只要查看改變是否有流即可

#include <cstdio>
#include <queue>
#include <algorithm>
#include <cstring>
using namespace std;
const int N=1005,INF=1<<29;
int m,n,x,sum;
struct node {
	int nxt,ro,to,flow;
}e[78000];
int dis[N];
int cur[N];
bool vis[N];
int head[N],tot=2;
inline void add(int u,int v,int ro)
{
	e[tot].to=v,e[tot].ro=ro,e[tot].nxt=head[u],head[u]=tot,tot++;
	e[tot].to=u,e[tot].ro=0,e[tot].nxt=head[v],head[v]=tot,tot++;
}
inline bool bfs(int s,int t)
{
	register int i;
	memset(vis,0,sizeof(vis));
	queue<int> q;
	dis[s]=0;vis[s]=1;
	q.push(s);
	while (!q.empty())
	{
		int x=q.front();
		q.pop();
		for (i=head[x];i;i=e[i].nxt)
		{
			if (!vis[e[i].to]&&e[i].ro>e[i].flow)
			{
				q.push(e[i].to);
				vis[e[i].to]=1;
				dis[e[i].to]=dis[x]+1;
			}
		}
	}
	return vis[t];
}
int s,t;
inline int dfs(int x,int a)
{
	if (x==t||a==0) return a;
	int flow=0,f;
	for (int& i=cur[x];i;i=e[i].nxt)
	{
		if (dis[x]+1==dis[e[i].to]&&((f=dfs(e[i].to,min(a,e[i].ro-e[i].flow)))>0))
		{
			e[i].flow+=f;
			e[i^1].flow-=f;
			flow+=f;
			a-=f;
			if (a==0) break;
		}
	}
	return flow;
}
inline int dinic(int s,int t)
{
	int flow=0;
	while (bfs(s,t))
	{
		for (int i=0;i<=n+m+1;i++) cur[i]=head[i];
		flow+=dfs(s,INF);
	}
	return flow;
}

int main()
{
	register int i,j;
	scanf("%d %d",&m,&n);
	s=0;t=n+m+1;
	//memset(head,-1,sizeof(head));
	for (i=1;i<=m;i++) scanf("%d",&x),add(s,i,x),sum+=x;
	for (i=1;i<=n;i++)
	{
		scanf("%d",&x);
		for (j=1;j<=m;j++) add(j,i+m,1);
		add(i+m,t,x);
	}
	int mf=dinic(s,t);
	if (mf<sum) printf("0");
	else 
	{
		puts("1");
		for (i=1;i<=m;i++)
		{
			for (j=head[i];j;j=e[j].nxt)
			{
				if (e[j].flow==1) printf("%d ",e[j].to-m);
			}
			printf("\n");
		}
	}
}