洛谷·[网络流24题]运输问题

初见安~这里是传送门：洛谷P4015 运输问题

题解

因为货物的初始量和变化后的量都是确定的，所以这个是最大/最小费用流。

其实直接按题目意思建图就可以了。从S连向每个a仓库表示分配货物，从每个b连向T表示限制b商店需要的货物数量。a和b之间就按照题意连费用流的边即可。

对于这个题的两个文我们当然可以写两个spfa，一个跑最大费用流，一个跑最小费用流。但是其实对于最大费用流，我们可以把边权全部取相反数然后跑最小费用流，再将结果取相反数。这样的话可以方便很多。【可能是我自己的问题，我把求最小费用流的时候的边权取相反数然后扔最大流跑就出不来了QAQ】

上代码——

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
#define maxn 500
#define maxm 200005
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
int read() {
	int x = 0, f = 1, ch = getchar();
	while(!isdigit(ch)) {if(ch == '-') f = -1; ch = getchar();}
	while(isdigit(ch)) x = (x << 1) + (x << 3) + ch - '0', ch = getchar();
	return x * f;
}

struct edge {int to, w, f, nxt;} e[maxm], e2[maxm];
int head[maxn], k = 0;
void add(int u, int v, int w, int f) {
	e[k] = {v, w, f, head[u]}; head[u] = k++;
	e[k] = {u, 0, -f, head[v]}; head[v] = k++;
} 

int S, T;
int dis[maxn], pre[maxn], incf[maxn];
bool vis[maxn];
bool spfa() {
	memset(dis, 0x3f, sizeof dis); dis[S] = 0;
	memset(vis, 0, sizeof vis); vis[S] = true;
	memset(incf, 0x3f, sizeof incf); incf[S] = INF; 
	queue<int> q; q.push(S);
	while(q.size()) {
		register int u = q.front(), v; q.pop(); vis[u] = false;
		for(int i = head[u]; ~i; i = e[i].nxt) {
			v = e[i].to; if(e[i].w && dis[u] + e[i].f < dis[v]) {
				dis[v] = dis[u] + e[i].f;
				incf[v] = min(incf[u], e[i].w); pre[v] = i;
				if(!vis[v]) q.push(v), vis[v] = true;
			}
		}
	}
	if(dis[T] == INF) return false; return true;
}

ll ans = 0;
void update() {
	register int u = T, i;
	while(u != S) {
		i = pre[u];
		e[i].w -= incf[T], e[i ^ 1].w += incf[T];
		u = e[i ^ 1].to;
	}
	ans += 1ll * incf[T] * dis[T];
}

int n, m, a[maxn], b[maxn]; 
signed main() {
	memset(head, -1, sizeof head);
	n = read(), m = read();
	S = 0, T = n + m + 1;//↓按照提议建图
	for(int i = 1, x; i <= n; i++) a[i] = read(), add(S, i, a[i], 0);
	for(int i = 1, x; i <= m; i++) b[i] = read(), add(i + n, T, b[i], 0);
	for(int i = 1; i <= n; i++) for(int j = 1, x; j <= m; j++)
		x = read(), add(i, j + n, INF, x);
	
	for(int i = 0; i < k; i++) e2[i] = e[i];//e2存放一下原图
	while(spfa()) update();
	printf("%lld\n", ans); ans = 0;
	
	for(int i = 0; i < k; i++) e[i] = e2[i], e[i].f = -e[i].f;//把原图换回来
	while(spfa()) update();
	printf("%lld\n", -ans);
	return 0;
}

迎评：）
——End——