Nu aveti permisiuni pentru a descarca fisierul grader_test1.in
Cod sursa(job #3298364)
Utilizator | Data | 29 mai 2025 09:45:09 | |
---|---|---|---|
Problema | Flux maxim | Scor | 100 |
Compilator | cpp-64 | Status | done |
Runda | Arhiva educationala | Marime | 1.92 kb |
#include <iostream>
#include <vector>
#include <algorithm>
#include <cassert>
#include <queue>
const int INF = +1e9;
struct Dinic {
struct Edge {
int u, v, cap;
};
std::vector<std::vector<int>> g;
std::vector<Edge> e;
std::vector<int> dist;
void init(int n) {
g.resize(n + 1);
dist.resize(n + 1);
}
void add_edge(int u, int v, int cap) {
g[u].push_back((int) e.size());
e.push_back({u, v, cap});
g[v].push_back((int) e.size());
e.push_back({v, u, 0});
}
bool can_bfs(int S, int T) {
std::queue<int> q;
dist.assign((int) dist.size(), INF);
dist[S] = 0;
q.push(S);
while (!q.empty()) {
int u = q.front();
q.pop();
for (const auto &id : g[u]) {
auto [uu, v, cap] = e[id];
if (cap != 0 && dist[u] + 1 < dist[v]) {
dist[v] = 1 + dist[u];
q.push(v);
}
}
}
return dist[T] != INF;
}
int push_flow(int u, int T, int F) {
if (F == 0 || u == T) {
return F;
}
int ret = 0;
for (const auto &id : g[u]) {
auto [uu, v, cap] = e[id];
if (dist[u] + 1 == dist[v]) {
int push = push_flow(v, T, std::min(F, cap));
F -= push;
ret += push;
e[id].cap -= push;
e[id ^ 1].cap += push;
}
}
return ret;
}
int max_flow(int S, int T) {
int ret = 0;
for (int jump = (1 << 16); jump > 0; jump >>= 1) {
while (can_bfs(S, T)) {
ret += push_flow(S, T, jump);
}
}
return ret;
}
};
int main() {
#ifndef LOCAL
freopen("maxflow.in", "r", stdin);
freopen("maxflow.out", "w", stdout);
#endif
int n, m;
std::cin >> n >> m;
Dinic F;
F.init(n);
for (int i = 0; i < m; i++) {
int u, v, c;
std::cin >> u >> v >> c;
F.add_edge(u, v, c);
}
std::cout << F.max_flow(1, n);
return 0;
}