运输问题(费用流 )

https://www.acwing.com/problem/content/2194/

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=5010,M=200010,inf=1e8;
int S,T;
int h[N],e[M],f[M],w[M],ne[M],idx;
int q[N],d[N],pre[N],incf[N];
bool st[N];
void add(int a,int b,int c,int d)
{
	e[idx]=b;f[idx]=c;w[idx]=d;ne[idx]=h[a];h[a]=idx++;
	e[idx]=a;f[idx]=0;w[idx]=-d;ne[idx]=h[b];h[b]=idx++;
}
bool spfa()
{
	int hh=0,tt=1;
	memset(d,0x3f,sizeof d);
	memset(incf,0,sizeof incf);
	q[0]=S;d[S]=0,incf[S]=inf;
	while(hh!=tt)
	{
		int t=q[hh++];
		if(hh==N) hh=0;
		st[t]=0;
		for(int i=h[t];i!=-1;i=ne[i])
		{
			int ver=e[i];
			if(f[i]&&d[ver]>d[t]+w[i])
			{
				d[ver]=d[t]+w[i];
				pre[ver]=i;
				incf[ver]=min(f[i],incf[t]);
				if(!st[ver])
				{
					q[tt++]=ver;
					if(tt==N) tt=0;
					st[ver]=1;
				}
			}
		}
	}
	return incf[T]>0;
}
void EK(int &flow,int &cost)
{
	flow=cost=0;
	while(spfa())
	{
		int t=incf[T];
		flow+=t,cost+=t*d[T];
		for(int i=T;i!=S;i=e[pre[i]^1])
		{
			f[pre[i]]-=t;
			f[pre[i]^1]+=t;
		}
	}
}
int main()
{
	int n,m;
	scanf("%d%d",&m,&n);
	memset(h,-1,sizeof h);
	S=0,T=m+n+1;
	for(int i=1;i<=m;i++)
	{
		int a;
		scanf("%d",&a);
		add(0,i,a,0);
	}
	for(int i=1;i<=n;i++)
	{
		int a;
		scanf("%d",&a);
		add(i+m,m+n+1,a,0);
	}
	for(int i=1;i<=m;i++)
	{
		for(int j=1;j<=n;j++)
		{
			int c;
			scanf("%d",&c);
			add(i,j+m,inf,c);
		}
	}	
	int flow,cost;
	EK(flow,cost);
	printf("%d\n",cost);
	for(int i=0;i<idx;i+=2)
	{
		f[i]+=f[i^1];f[i^1]=0;//恢复原网络 
		w[i]=-w[i];w[i^1]=-w[i^1];
	}
	EK(flow,cost);
	printf("%d\n",-cost);
	return 0;
}