洛谷P1344 追查坏牛奶Pollutant Control

题目描述

你第一天接手三鹿牛奶公司就发生了一件倒霉的事情：公司不小心发送了一批有三聚氰胺的牛奶。很不幸，你发现这件事的时候，有三聚氰胺的牛奶已经进入了送货网。这个送货网很大，而且关系复杂。你知道这批牛奶要发给哪个零售商，但是要把这批牛奶送到他手中有许多种途径。送货网由一些仓库和运输卡车组成，每辆卡车都在各自固定的两个仓库之间单向运输牛奶。在追查这些有三聚氰胺的牛奶的时候，有必要保证它不被送到零售商手里，所以必须使某些运输卡车停止运输，但是停止每辆卡车都会有一定的经济损失。你的任务是，在保证坏牛奶不送到零售商的前提下，制定出停止卡车运输的方案，使损失最小。

输入格式

第一行: 两个整数N(2<=N<=32)、M(0<=M<=1000), N表示仓库的数目，M表示运输卡车的数量。仓库1代 表发货工厂，仓库N代表有三聚氰胺的牛奶要发往的零售商。 第2..M+1行: 每行3个整数Si,Ei,Ci。其中Si,Ei表示这 辆卡车的出发仓库，目的仓库。Ci(0 <= C i <= 2,000,000) 表示让这辆卡车停止运输的损失。

输出格式

两个整数C、T：C表示最小的损失，T表示在损失最小的前提下，最少要停止的卡车数。

输入输出样例

输入 #1复制

4 5
1 3 100
3 2 50
2 4 60
1 2 40
2 3 80

输出 #1复制

60 1

说明/提示

题目翻译来自NOCOW。

USACO Training Section 4.4

上代码：

#include<bits/stdc++.h>
#define inf 2147483647
using namespace std;
int head[33], ver[2001], edge[2001], nxt[2001], sz = 1;
void addedge(int u, int v, int w) {
	ver[++sz] = v, edge[sz] = w, nxt[sz] = head[u], head[u] = sz;
	ver[++sz] = u, edge[sz] = 0, nxt[sz] = head[v], head[v] = sz;
}
int cur[33];
long long dis[2001];
bool bfs(int s, int t) {
	memset(dis, 0x3f, sizeof(dis));
	memcpy(cur, head, sizeof(cur));
	queue<int> q;
	q.push(s);
	dis[s] = 0;
	while (q.size()) {
		int u = q.front();
		q.pop();
		for (int i = head[u]; i; i = nxt[i])
			if (edge[i] && dis[ver[i]] > dis[u] + 1) {
				dis[ver[i]] = dis[u] + 1;
				q.push(ver[i]);
			}
	}
	return dis[t] < 0x3f3f3f3f;
}
int dinic(int x, int t, int flow) {
	if (x == t) return flow;
	int f0 = flow;
	for (int i = cur[x]; i && flow; cur[x] = i, i = nxt[i])
		if (edge[i] && dis[ver[i]] == dis[x] + 1) {
			int f = dinic(ver[i], t, min(flow, edge[i]));
			edge[i] -= f, edge[i^1] += f;
			flow -= f;
		}
	return f0 - flow;
}
long long maxflow;
int main() {
	int n, m;
	cin >> n >> m;
	addedge(1, n, 0);
	for (int i = 1; i <= m; i++) {
		int s, e, c;
		cin >> s >> e >> c;
		addedge(s, e, c*(m+1)+1);
	}
	while (bfs(1, n)) maxflow += dinic(1, n, 2147483647);
	cout << maxflow / (m+1) << ' ' << maxflow%(m+1) << endl;
	for (int i = 4; i <= m * 2 + 2; i += 2) {
		if (edge[i]) continue;
		int tmp = edge[i^1];
		int tot = edge[i^1]; edge[i^1] = 0;
		int u = ver[i^1], v = ver[i];
		edge[2] = inf;
		while (tot) bfs(u, v), tot -= dinic(u, v, tot);
		int used = inf - edge[2];
		edge[2] = 0;
		if (used < tmp || bfs(1, n)) {
			edge[3] = used;
			int tot = tmp;
			while(tot) bfs(v, u), tot -= dinic(v, u, tot);
			while (edge[3]) bfs(1, n), edge[3] -= dinic(1, n, edge[3]);
			edge[i^1] = tmp;
			continue;
		}
		cout << i/2-1 << endl;
	}
}