2236. 伊基的故事 I - 道路重建（最大流）

活动 - AcWing

伊基是一个小国 – 凤凰国的国王。

凤凰国是如此之小，以至于只有一个城市负责日常商品的生产，并使用公路网将商品运送到首都。

伊基发现本国最大的问题在于运输速度太慢了。

因为伊基以前是 ACM/ICPC 的参赛者，他意识到这其实是一个最大流问题。

他编写了一个最大流程序，并计算出了当前运输网络的最大运输能力。

他对运输速度的现状十分不满，并希望能够提高国家的运输能力。

提高运输能力的方法很简单，伊基将在运输网络中重建一些道路，以使这些道路具有更高的运输能力。

但是不幸的是，凤凰国的财力有限，道路建设经费只够重建一条道路。

伊基想要知道共有多少条道路可以纳入重建道路候选名单。

这些道路需要满足，将其重建后，国家的总运输能力能够增加。

输入格式

第一行包含 N 和 M，分别表示城市和道路的数量。

接下来 M 行，每行包含三个整数 a,b,c，表示存在一条道路从城市 a 通往城市 b，且运输能力为 c。

所有道路都是有方向的。

城市编号从 0 到 N−1。

生产日常商品的城市为 0 号城市，首都为 N−1 号城市。

输出格式

输出一个整数 K，表示存在 K 条道路，对其中每条道路进行重建都会增加运输网络的运输能力。

数据范围

1≤N≤500,
1≤M≤5000,
0≤a,b<N
0≤c≤100

输入样例：

2 1
0 1 1

输出样例：

1

解析： 

性质：侯选边（u,v）在残留网络中必须是满流及容量为零且从源点 S 出发的流量能到达 u，从 v 出发的流量能到达汇点 T。

#include<iostream>
#include<string>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<utility>
#include<stack>
#include<queue>
#include<vector>
#include<set>
#include<math.h>
#include<map>
#include<sstream>
#include<deque>
#include<unordered_map>
#include<unordered_set>
#include<bitset>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PII;
const int N = 5e2 + 10, M = 1e4+5, INF = 0x3f3f3f3f;
int n, m, S, T;
int h[N], e[M], f[M], ne[M], idx;
int q[N], d[N], cur[N];
bool vis_s[N], vis_t[N];

void add(int a, int b, int c) {
	e[idx] = b, f[idx] = c, ne[idx] = h[a], h[a] = idx++;
	e[idx] = a, f[idx] = 0, ne[idx] = h[b], h[b] = idx++;
}

bool bfs() {
	int hh = 0, tt = 0;
	memset(d, -1, sizeof d);
	q[0] = S, d[S] = 0, cur[S] = h[S];
	while (hh <= tt) {
		int t = q[hh++];
		for (int i = h[t]; i != -1; i = ne[i]) {
			int j = e[i];
			if (d[j] == -1 && f[i]) {
				cur[j] = h[j];
				d[j] = d[t] + 1;
				if (j == T)return 1;
				q[++tt] = j;
			}
		}
	}
	return 0;
}

int find(int u, int limit) {
	if (u == T)return limit;
	int flow = 0;
	for (int i = cur[u]; i != -1 && flow < limit; i = ne[i]) {
		int j = e[i];
		cur[u] = i;
		if (d[j] == d[u] + 1 && f[i]) {
			int t = find(j, min(f[i], limit - flow));
			if (!t)d[j] = -1;
			f[i] -= t, f[i ^ 1] += t, flow += t;
		}
	}
	return flow;
}

int dinic() {
	//cout << "LLLLLLLLLLLLLLLLLLLLLLLLLLLLL" << endl;
	int ret = 0, flow;
	while (bfs())while (flow = find(S, INF))ret += flow;
	return ret;
}

void dfs(int u, bool st[], int x) {
	st[u] = 1;
	for (int i = h[u]; i != -1; i = ne[i]) {
		int j = i ^ x, ver = e[i];
		if (f[j] && !st[ver])
			dfs(ver, st, x);
	}
}

int main() {
	cin >> n >> m;
	memset(h, -1, sizeof h);
	S = 0, T = n - 1;
	for (int i = 1, a, b, c; i <= m; i ++) {
		scanf("%d%d%d", &a, &b, &c);
		add(a, b, c);
	}
	//cout << "KKKKKKKKKKKKKKKKKKKKKKKKKKKKKK" << endl;
	int t = dinic();
	//cout << t << endl;
	dfs(S, vis_s, 0);
	dfs(T, vis_t, 1);
	int ret = 0;
	for (int i = 0; i < m * 2; i += 2)
		if (vis_t[e[i]] && vis_s[e[i ^ 1]]&&!f[i])
			ret += 1;
	cout << ret << endl;
	return 0;
}