CodeForces - 546E Soldier and Traveling（最大流+拆点+建图+输出流量流向）

链接：https://cn.vjudge.net/problem/CodeForces-546E

题意：n个城市，每个城市初始时有a[i]个士兵。城市间有m条无向边，某个城市的士兵可以留在原城市到或到与该城市仅仅相邻的城市（即只隔一条路的士兵）。现在要求最终某个城市必须正好有b[i]个士兵，可能的话输出n*n的矩阵，表示有mat[i][j]个士兵从城市i到城市j。

思路：因为城市有初始数量和目标数量，有一个点有两个相关值得话，就去想拆点。

1、城市（i）和源点（s）连边，容量为a[i]。

2、m条边的两个点之间连边，u和v+n，u+n和v；容量为inf。

3.城市（拆点：i+n）与汇点（t）连边，容量为b[i]。

跑一遍最大流即可。

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int N = 2e2+10;
const int M = 4e4+10;
const int inf = 0x3f3f3f3f;
struct node
{
	int to,ca,nxt;
	node(){}
	node(int to,int ca,int nxt):to(to),ca(ca),nxt(nxt){}
	
}g[M];
int head[N],cnt;
int deep[N],cur[N];
int s,t,suma,sumb;
int n,m,u,v;
int a[N],b[N],mp[N][N];
void Init()
{
	cnt=0;
	s=0; t=n<<1|1;
	for(int i=s;i<=t;i++)
		head[i]=-1;
}
void add(int u,int v,int ca)
{
	g[cnt]=node{v,ca,head[u]},head[u]=cnt++;
}
bool bfs()
{
	queue<int> q;
	for(int i=s;i<=t;i++)
		deep[i]=0;
	deep[s]=1;
	q.push(s);
	while(!q.empty())
	{
		u=q.front();
		q.pop();
		if(u==t) return 1;
		for(int i=head[u];~i;i=g[i].nxt)
		{
			v=g[i].to;
			if(!deep[v]&&g[i].ca>0)
			{
				deep[v]=deep[u]+1;
				q.push(v);	
			}	
		}	
	}	
	return deep[t]!=0;
}
int getid(int x)
{
	return x>n?x-n:x;
}
int dfs1(int u,int flow)
{
	if(u==t||!flow) return flow;
	int ans=0,nowflow;
	for(int& i=cur[u];~i;i=g[i].nxt)
	{
		v=g[i].to;
		if(deep[v]==deep[u]+1&&g[i].ca>0)
		{
			nowflow=dfs1(v,min(flow,g[i].ca));
			if(nowflow)
			{
				ans+=nowflow;
				flow-=nowflow;
				g[i].ca-=nowflow;
				g[i^1].ca+=nowflow;
				if(!flow) break;
			}
		}
	}
	if(!ans) deep[u]=0;
	return ans;
}
int dfs(int u,int flow)
{
	if(u==t||!flow) return flow;
	int v,nowflow;
	for(int i=head[u];~i;i=g[i].nxt)
	{
		v=g[i].to;
		if(deep[v]==deep[u]+1&&g[i].ca>0)
		{
			nowflow=dfs(v,min(flow,g[i].ca));
			if(nowflow)
			{
				g[i].ca-=nowflow;
				g[i^1].ca+=nowflow;
				return nowflow;	
			} 	
		}	
	} 
	return 0;
}
void dinic()
{
	int maxflow=0,flow;
	while(bfs())
	{
		memcpy(cur,head,sizeof(head));
		while(flow=dfs1(s,inf))
			maxflow+=flow;
	}
	if(maxflow!=sumb)
	{
		puts("NO");
		return ;
	}
	puts("YES");
	for(u=1;u<=n;u++)
		for(int i=head[u];~i;i=g[i].nxt)
			if(g[i].to>n&&g[i^1].ca>0)
				mp[u][g[i].to-n]=g[i^1].ca;
			
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			printf("%d%c",mp[i][j]," \n"[j==n]);
}
int main(void)
{
	suma=0,sumb=0;
	scanf("%d%d",&n,&m);
	Init(); 
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]),suma+=a[i];
	for(int i=1;i<=n;i++)
		scanf("%d",&b[i]),sumb+=b[i];
	if(suma!=sumb)
	{
		puts("NO");
		return 0;	
	}	
	for(int i=1;i<=n;i++)
	{
		add(s,i,a[i]);
		add(i,s,0);
		
		add(i,i+n,inf);
		add(i+n,i,0);
		
		add(i+n,t,b[i]);
		add(t,i+n,0);		
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&u,&v);
		add(u,v+n,inf);
		add(v+n,u,0);
		
		add(v,u+n,inf);
		add(u+n,v,0);				
	}
	dinic();
	return 0;
}