-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathdinic.cpp
94 lines (81 loc) · 2.06 KB
/
dinic.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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
// added
#include <bits/stdc++.h>
using namespace std;
using i64 = int64_t;
using vi = vector<i64>;
using vvi = vector<vi>;
class Dinic {
struct edge {
i64 to, cap, rev;
};
int n;
vector<vector<edge>> G;
vi level;
vi iter;
void bfs(int s) {
level.assign(n, -1);
queue<int> que;
que.push(s);
level[s] = 0;
while (que.size()) {
int v = que.front();
que.pop();
for (int i = 0; i < G[v].size(); i++) {
edge &e = G[v][i];
if (e.cap > 0 && level[e.to] < 0) {
level[e.to] = level[v] + 1;
que.push(e.to);
}
}
}
}
i64 dfs(int v, int t, i64 f) {
if (v == t) return f;
for (int i = iter[v]; i < G[v].size(); i++) {
edge &e = G[v][i];
if (e.cap > 0 && level[v] < level[e.to]) {
i64 d = dfs(e.to, t, min(f, e.cap));
if (d > 0) {
e.cap -= d;
G[e.to][e.rev].cap += d;
return d;
}
}
}
return 0;
}
public:
Dinic(int n) : n(n), G(n), level(n), iter(n) {}
void addEdge(int from, int to, i64 cap) {
G[from].push_back({to, cap, int(G[to].size())});
G[to].push_back({from, 0, int(G[from].size()) - 1});
}
i64 max_flow(int s, int t) {
i64 flow = 0;
while (1) {
bfs(s);
if (level[t] < 0) return flow;
iter.assign(n, 0);
i64 f;
while ((f = dfs(s, t, 1e18)) > 0) {
flow += f;
}
}
}
};
int main() {
int n, m, p, g;
cin >> n >> m >> p >> g;
Dinic d(m + 1);
for (int i = 0; i < g; i++) {
int l;
cin >> l;
d.addEdge(m, l, 1e9);
}
for (int i = 0; i < n; i++) {
int from, to, cap;
cin >> from >> to >> cap;
d.addEdge(from, to, cap);
}
cout << (d.max_flow(m, 0) >= p ? "Yes" : "No") << endl;
}