Cod sursa(job #2689566)

Utilizator nicu_ducalNicu Ducal nicu_ducal Data 21 decembrie 2020 13:17:12
Problema Flux maxim Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 2.98 kb
#include <bits/stdc++.h>

using namespace std;

template<typename T>
ostream &operator<<(ostream &os, const vector<T> &v) {
  os << '{';
  string sep;
  for (const auto &x : v) os << sep << x, sep = ", ";
  return os << '}';
}

template<typename A, typename B>
ostream &operator<<(ostream &os, const pair<A, B> &p) { return os << '(' << p.first << ", " << p.second << ')'; }

using i64 = long long int;

const int INF = INT_MAX, MOD = 1e9 + 7;
const double EPS = 1e-9, PI = acos(-1);
const int dx[] = {0, 0, 0, -1, 1, -1, 1, 1, -1};
const int dy[] = {0, -1, 1, 0, 0, -1, 1, -1, 1};

/*
 * Algorithm: Dinic's Algorithm (Complexity: O(|V|^2 * |E|)
 */
struct FlowEdge {
    long long u, v;
    long long cap, flow = 0;

    FlowEdge(long long u, long long v, long long cap) : v(v), u(u), cap(cap) {}
};

struct Graph {
    const long long INF_FLOW = 1e18;
    vector<FlowEdge> edges;
    vector<vector<int>> adj;
    long long n, m = 0;
    long s, t;
    vector<int> level, ptr;
    queue<int> q;

    Graph(int _n = 0) {
      init(_n);
    }

    void init(int _n) {
      n = _n;
      s = 1, t = n;
      adj.resize(n + 1);
      level.resize(n + 1, 0);
      ptr.resize(n + 1, 0);
    }

    void addEdge(int u, int v, long long c) {
      edges.emplace_back(u, v, c);
      edges.emplace_back(v, u, 0);
      adj[u].push_back(m);
      adj[v].push_back(m + 1);
      m += 2;
    }

    bool bfs() {
      while (not q.empty()) {
        int u = q.front();
        q.pop();
        for (auto idx: adj[u]) {
          if (edges[idx].cap - edges[idx].flow < 1 or level[edges[idx].v] != -1) continue;
          level[edges[idx].v] = level[u] + 1;
          q.push(edges[idx].v);
        }
      }
      return (level[t] != -1);
    }

    long long dfs(int u, long long pushed) {
      if (pushed == 0) return 0;
      if (u == t) return pushed;
      for (int &cid = ptr[u]; cid < (int) adj[u].size(); cid++) {
        int idx = adj[u][cid];
        int v = edges[idx].v;
        if (level[u] + 1 != level[v] or edges[idx].cap - edges[idx].flow < 1) continue;
        long long tr = dfs(v, min(pushed, edges[idx].cap - edges[idx].flow));
        if (tr == 0) continue;
        edges[idx].flow += tr;
        edges[idx ^ 1].flow -= tr;
        return tr;
      }
      return 0;
    }

    long long flow() {
      long long fw = 0;
      while (true) {
        fill(level.begin(), level.end(), -1);
        level[s] = 0;
        q.push(s);
        if (not bfs()) break;
        fill(ptr.begin(), ptr.end(), 0);
        while (long long pushed = dfs(s, INF_FLOW))
          fw += pushed;
      }
      return fw;
    }
};

int main() {
  ios_base::sync_with_stdio(false);
  cin.tie(nullptr);
  cout.tie(nullptr);
  /// mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
  ifstream cin("maxflow.in");
  ofstream cout("maxflow.out");

  int n, m;
  cin >> n >> m;
  Graph graph(n);
  for (int i = 1; i <= m; i++) {
    i64 u, v, c;
    cin >> u >> v >> c;
    graph.addEdge(u, v, c);
  }
  cout << graph.flow() << "\n";

  return 0;
}