-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDinic.cpp
More file actions
54 lines (54 loc) · 1.61 KB
/
Dinic.cpp
File metadata and controls
54 lines (54 loc) · 1.61 KB
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
struct edge {
int to, cap, rev;
};
vector<edge> g[101010];
static const int INF = 0x3f3f3f3f;
int level[101010];
int iter[101010];
void add_edge(int from, int to, int cap) {
g[from].push_back((edge) { to, cap, (int) g[to].size() });
g[to].push_back((edge) { from, 0, (int) g[from].size() - 1 });
}
void bfs(int s) {
memset(level, -1, sizeof level);
queue<int> q;
level[s] = 0;
q.push(s);
while (!q.empty()) {
int v = q.front();
q.pop();
for (edge &e : g[v]) {
if (e.cap > 0 && level[e.to] < 0) {
level[e.to] = level[v] + 1;
q.push(e.to);
}
}
}
}
int dfs(int v, int t, int 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]) {
int 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;
}
int MaxFlow(int s, int t) {
int flow = 0;
for (;;) {
bfs(s);
if (level[t] < 0) return flow;
memset(iter, 0, sizeof iter);
int f;
while ((f = dfs(s, t, INF)) > 0) {
flow += f;
}
}
}