-
Notifications
You must be signed in to change notification settings - Fork 4
/
Copy path1694.cpp
70 lines (64 loc) · 1.32 KB
/
1694.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#include <bits/stdc++.h>
using namespace std;
const int N = 505;
struct Edge {
int v, c;
};
int n, m, lev[N], ptr[N];
vector<Edge> edges;
vector<int> g[N];
queue<int> q;
void add_edge(int u, int v, int c) {
g[u].emplace_back(edges.size());
edges.push_back({v, c});
g[v].emplace_back(edges.size());
edges.push_back({u, 0});
}
bool bfs() {
fill(lev + 1, lev + 1 + n, -1);
lev[1] = 0;
q.emplace(1);
while (!q.empty()) {
int u = q.front(); q.pop();
for (int id: g[u]) {
auto [v, c] = edges[id];
if (lev[v] == -1 && c > 0) {
lev[v] = lev[u] + 1;
q.emplace(v);
}
}
}
return (lev[n] != -1);
}
int dfs(int u, int pushed) {
if (u == n) return pushed;
while (ptr[u] < (int)g[u].size()) {
int id = g[u][ptr[u]++];
auto [v, c] = edges[id];
if (lev[v] == lev[u] + 1 && c > 0) {
int t = dfs(v, min(pushed, c));
if (t) {
edges[id].c -= t;
edges[id ^ 1].c += t;
return t;
}
}
}
return 0;
}
int main() {
ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
cin >> n >> m;
for (int u, v, c; m--;) {
cin >> u >> v >> c;
add_edge(u, v, c);
}
long long ans = 0;
while (true) {
if (!bfs()) break;
fill(ptr + 1, ptr + 1 + n, 0);
while (int t = dfs(1, INT_MAX)) ans += t;
}
cout << ans;
return 0;
}