网络流二十四题 ————(十四）、P4015 运输问题 费用流

费用流最简单的模型。

容量控制选择权重，费用控制选择代价。

最小费用最大流，费用加负数就是最大费用最大流。

#include<bits/stdc++.h> 
using namespace std;
const int MAXN = 5001;
const int MAXM = 50001;
struct MCMF
{	
	int n,  s, t, cnt = 1;
	long long maxflow=0, mincost=0;
	int dis[MAXN], head[MAXN], incf[MAXN], pre[MAXN];//dis表示最短路，incf表示当前增广路上最小流量，pre表示前驱
	bool vis[MAXN];
	struct EDGE {
		int next, to, dis, flow;
	}ee[MAXM << 1];
	inline void AD(int from, int to, int flow, int dis) {
		ee[++cnt].next = head[from];
		ee[cnt].to = to;
		ee[cnt].dis = dis;
		ee[cnt].flow = flow;
		head[from] = cnt;
	}
	inline void add(int x, int y, int flow, int dis){
		AD(x,y,flow,dis);AD(y,x,0,-dis);
		//双向边 
	}
	inline bool spfa() 
	{
		queue <int> q;
		for(int i=1;i<=n;i++)dis[i]=0x3f3f3f3f,vis[i]=0; 
		q.push(s);
		dis[s] = 0;vis[s] = 1;
		incf[s] = 1 << 30;
		while(!q.empty()) {
			int u = q.front();
			vis[u] = 0;
			q.pop();
			for(register int i = head[u]; i; i = ee[i].next) {
				if(!ee[i].flow) continue;//没有剩余流量
				int v = ee[i].to;
				if(dis[v] > dis[u] + ee[i].dis) {
					dis[v] = dis[u] + ee[i].dis;
					incf[v] = min(incf[u], ee[i].flow);//更新incf
					pre[v] = i;
					if(!vis[v]) vis[v] = 1, q.push(v);
				}
			}
		}
		if(dis[t] == 0x3f3f3f3f) return 0;
		return 1;
	}
	void gao() 
	{
		while(spfa()) {//如果有增广路
			int x = t;
			maxflow += incf[t];
			mincost += (long long)dis[t] * incf[t];
			int i;
			while(x != s) {//遍历这条增广路，正向边减流反向边加流
				i = pre[x];
				ee[i].flow -= incf[t];
				ee[i^1].flow += incf[t];
				x = ee[i^1].to;
			}
		}
	}
	//先初始化 
	void init(int nn,int S,int T)
	{
		n=nn,s=S,t=T;
		cnt=1;
		for(int i=0;i<=n;i++)head[i]=0;
		mincost=maxflow=0;
	}
}mc;
int a[110],b[110],c[110][110];
int main() 
{
	int n,m,s,t,x;
	cin>>n>>m;
	int N=n+m+2;
	s=N-1,t=N;
	mc.init(N,s,t);
	for(int i=1;i<=n;i++){
		cin>>x;a[i]=x;
		mc.add(s,i,x,0);
	}
	for(int i=1;i<=m;i++){
		cin>>x;b[i]=x;
		mc.add(i+n,t,x,0);
	}
	int inf = 0x3f3f3f3f;
	for(int i=1;i<=n;i++)
	for(int j=1;j<=m;j++){
		cin>>x;c[i][j]=x;
		mc.add(i,j+n,inf,x);
	} 
	mc.gao();
	cout<<mc.mincost<<endl;
	mc.init(N,s,t);
	for(int i=1;i<=n;i++){
		x=a[i];
		mc.add(s,i,x,0);
	}
	for(int i=1;i<=m;i++){
		x=b[i];
		mc.add(i+n,t,x,0);
	}
	for(int i=1;i<=n;i++)
	for(int j=1;j<=m;j++){
		x=c[i][j];
		mc.add(i,j+n,inf,-x);
	} 
	mc.gao();
	cout<<-mc.mincost<<endl;
	return 0;
}