[网络流24题] 洛谷P4012 深海机器人问题 费用流

给出$P\times Q$的网格，$P,Q\le 15$，以及$a\le 4$组机器人起点所在的位置，以及$b\le 6$组机器人终点所在的位置。机器人每次可以向上或者向右移动。并且给出每条边权上的物品的价值，但是这个物品只能被机器人取一次，求所有机器人获得物品的最大价值和。
也是费用流的基础题。
给网格图标号，源点连所有的出发位置的机器人机器人个数的容量和$0$的费用，每个机器人结束位置连汇点机器人个数的容量和$0$的费用，然后对于所有的合法连边都连上无限容量和$0$的费用，对于给出的物品则连上$1$的容量和其价值的费用。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf=0x3f3f3f3f;
namespace MCMF {
	const int N=405,M=3e4+3;
	int tot=-1,n,S,T; 
	int front[N],pre[N],lst[N];
	ll flow[N],dis[N];
	bool inq[N];
	struct edge{ int v,nxt;ll c,w; }e[2*M];
	void init() {
		tot=-1;
		for(int i=1;i<=n;i++) front[i]=-1;
	}
	void add(int u,int v,ll c,ll w) {
		e[++tot].v=v,e[tot].w=w,e[tot].c=c,e[tot].nxt=front[u],front[u]=tot;
		e[++tot].v=u,e[tot].w=-w,e[tot].c=0,e[tot].nxt=front[v],front[v]=tot;
	}
	void reset() {
		pre[T]=-1;
		for(int i=1;i<=n;i++) dis[i]=flow[i]=inf;
		for(int i=1;i<=n;i++) inq[i]=0;
	}
	bool spfa() {
		reset();
		queue<int> q;
		q.push(S);
		inq[S]=1;dis[S]=0;
		while(!q.empty()) {
			int u=q.front();
			q.pop();
			inq[u]=0;
			for(int i=front[u];i!=-1;i=e[i].nxt) {
				int v=e[i].v;
				if(e[i].c>0) {
					if(dis[v]>dis[u]+e[i].w) {
						dis[v]=dis[u]+e[i].w;
						flow[v]=min(flow[u],e[i].c);
						pre[v]=u,lst[v]=i;
						if(!inq[v]) {
							inq[v]=1;
							q.push(v);
						}
					}
				}
			}
		}
		return pre[T]!=-1?1:0; 
	}
	pair<ll,ll> solve() {
		ll maxflow=0,mincost=0;
		while(spfa()) {
			int cur=T;
			maxflow+=flow[T];
			mincost+=flow[T]*dis[T];
			while(cur!=S) {
				e[lst[cur]].c-=flow[T];
				e[lst[cur]^1].c+=flow[T];
				cur=pre[cur];
			}
		}
		return make_pair(maxflow,mincost);
	} 
} 
const int N=20;
int mp[N][N]; 
int main() {
	int a,b,P,Q,tot=0;
	scanf("%d%d%d%d",&a,&b,&P,&Q);
	MCMF::n=(P+1)*(Q+1)+2;
	int S=(P+1)*(Q+1)+1;
	int T=(P+1)*(Q+1)+2;
	MCMF::S=S;
	MCMF::T=T;	
	MCMF::init();
	for(int i=0;i<=P;i++) {
		for(int j=0;j<=Q;j++) {
			mp[i][j]=++tot;
		}
	}
	for(int i=0;i<=P;i++) {
		for(int j=0;j<=Q;j++) {
			if(i<P) MCMF::add(mp[i][j],mp[i+1][j],1e9,0);
			if(j<Q) MCMF::add(mp[i][j],mp[i][j+1],1e9,0); 
		}
	}
	for(int i=0;i<=P;i++) {
		for(int j=0;j<Q;j++) {
			int x;scanf("%d",&x);
			MCMF::add(mp[i][j],mp[i][j+1],1,-x);
		}
	}
	for(int i=0;i<=Q;i++) {
		for(int j=0;j<P;j++) {
			int x;scanf("%d",&x);
			MCMF::add(mp[j][i],mp[j+1][i],1,-x);
		}
	}
	for(int i=1;i<=a;i++) {
		int k,x,y;
		scanf("%d%d%d",&k,&y,&x);
		MCMF::add(S,mp[y][x],k,0); 
	}	
	for(int i=1;i<=b;i++) {
		int k,x,y;
		scanf("%d%d%d",&k,&y,&x);
		MCMF::add(mp[y][x],T,k,0);
	}
	printf("%lld\n",-MCMF::solve().second); 
	return 0;
}