Pagini recente » Cod sursa (job #1798025) | Cod sursa (job #1649439) | Cod sursa (job #441005) | Cod sursa (job #806953) | Cod sursa (job #984711)
Cod sursa(job #984711)
#include <algorithm>
#include <vector>
#include <climits>
#include <iostream>
#include <fstream>
using namespace std;
const int maxnodes = 5000;
int nodes = maxnodes, src, dest;
int dist[maxnodes], q[maxnodes], work[maxnodes];
struct Edge {
int to, rev;
int f, cap;
};
vector<Edge> g[maxnodes];
// Adds bidirectional edge
void addEdge(int s, int t, int cap){
Edge a = {t, (int)g[t].size(), 0, cap};
Edge b = {s, (int)g[s].size(), 0, 0};
g[s].push_back(a);
g[t].push_back(b);
}
bool dinic_bfs() {
fill(dist, dist + nodes, -1);
dist[src] = 0;
int qt = 0;
q[qt++] = src;
for (int qh = 0; qh < qt; qh++) {
int u = q[qh];
for (int j = 0; j < (int) g[u].size(); j++) {
Edge &e = g[u][j];
int v = e.to;
if (dist[v] < 0 && e.f < e.cap) {
dist[v] = dist[u] + 1;
q[qt++] = v;
}
}
}
return dist[dest] >= 0;
}
int dinic_dfs(int u, int f) {
if (u == dest)
return f;
for (int &i = work[u]; i < (int) g[u].size(); i++) {
Edge &e = g[u][i];
if (e.cap <= e.f) continue;
int v = e.to;
if (dist[v] == dist[u] + 1) {
int df = dinic_dfs(v, min(f, e.cap - e.f));
if (df > 0) {
e.f += df;
g[v][e.rev].f -= df;
return df;
}
}
}
return 0;
}
int maxFlow(int _src, int _dest) {
src = _src;
dest = _dest;
int result = 0;
while (dinic_bfs()) {
fill(work, work + nodes, 0);
while (int delta = dinic_dfs(src, INT_MAX))
result += delta;
}
return result;
}
int main() {
int n, m, a, b, c;
ifstream f("maxflow.in");
ofstream g("maxflow.out");
f >> n >> m;
for ( int i = 1; i <= m; ++i )
{
f >> a >> b >> c;
a--;
b--;
addEdge(a,b,c);
}
g << maxFlow(0,n-1) << endl;
}