Problem p02376 (C++) — buggy submission, verdict: Time Limit Exceeded

#include <bits/stdc++.h>

#define FOR(i, a, b) for (int i = (a); i < ((int)b); ++i)
#define RFOR(i, a) for (int i = (a); i >= 0; --i)
#define FOE(i, a) for (auto i : a)
#define ALL(c) (c).begin(), (c).end()
#define RALL(c) (c).rbegin(), (c).rend()
#define DUMP(x) cerr << #x << " = " << (x) << endl;
#define SUM(x) std::accumulate(ALL(x), 0LL)
#define MIN(v) *std::min_element(v.begin(), v.end())
#define MAX(v) *std::max_element(v.begin(), v.end())
#define EXIST(v, x) (std::find(v.begin(), v.end(), x) != v.end())
#define BIT(n) (1LL << (n))
#define UNIQUE(v) v.erase(unique(v.begin(), v.end()), v.end());

typedef long long LL;
template <typename T> using V = std::vector<T>;
template <typename T> using VV = std::vector<std::vector<T>>;
template <typename T> using VVV = std::vector<std::vector<std::vector<T>>>;
template <class T> inline T ceil(T a, T b) { return (a + b - 1) / b; }
template <class T> inline void print(T x) { std::cout << x << std::endl; }
template <class T> inline bool inside(T y, T x, T H, T W) {
  return 0 <= y and y < H and 0 <= x and x < W;
}
inline double distance(double y1, double x1, double y2, double x2) {
  return sqrt((x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2));
}

const int INF = 1L << 30;
const double EPS = 1e-9;
const std::string YES = "YES", Yes = "Yes", NO = "NO", No = "No";
const std::vector<int> dy = {0, 1, 0, -1},
                       dx = {1, 0, -1, 0}; // 4近傍（右, 下, 左, 上）

using namespace std;

// 最大流問題を解く O(|E||V|^2)
class Dinic {

  struct Edge {
    const unsigned int to;  // 行き先
    long long flow;         // 流量
    const long long cap;    // 容量
    const unsigned int rev; // 逆辺
    const bool is_rev;      // 逆辺かどうか

    Edge(unsigned int to, long long flow, long long cap, unsigned int rev,
         bool is_rev)
        : to(to), flow(flow), cap(cap), rev(rev), is_rev(is_rev) {
      assert(this->cap >= 0);
    }
  };

  vector<vector<Edge>> graph; // グラフの隣接リスト表現
  vector<int> level;          // sからの距離
  vector<int> iter;           // dfsがどこまで進んだか

public:
  Dinic(unsigned int num_of_node) {
    assert(num_of_node > 0);
    graph.resize(num_of_node);
    level.resize(num_of_node);
    iter.resize(num_of_node);
  }

  // fromからtoへ向かう容量capの辺をグラフに追加する
  void add_edge(unsigned int from, unsigned int to, long long cap) {
    graph[from].push_back(
        Edge(to, 0, cap, (unsigned int)graph[to].size(), false));
    graph[to].push_back(
        Edge(from, cap, cap, (unsigned int)graph[from].size() - 1, true));
  }

  // sからtへの最大流を求める
  long long max_flow(unsigned int s, unsigned int t) {
    long long flow = 0;
    while (true) {
      // 残余ネットワーク上でsから各ノードへの距離を求める
      bfs(s);
      if (level.at(t) < 0) {
        return flow;
      }
      // 距離が増加する向きの辺のみを使って流せるだけ流す
      fill(iter.begin(), iter.end(), 0);
      long long f;
      while ((f = dfs(s, t, INT_MAX)) > 0) {
        flow += f;
      }
    }
  }

private:
  // 残余ネットワーク上でsから全ノードへの最短距離をBFSで計算する
  void bfs(unsigned int s) {
    fill(level.begin(), level.end(), -1);
    queue<unsigned int> que;
    level.at(s) = 0;
    que.push(s);
    while (not que.empty()) {
      unsigned int v = que.front();
      que.pop();
      for (int i = 0; i < graph.at(v).size(); ++i) {
        Edge &e = graph.at(v).at(i);
        if (e.cap - e.flow > 0 && level.at(e.to) < 0) {
          level.at(e.to) = level.at(v) + 1;
          que.push(e.to);
        }
      }
    }
  }

  // vからtへf流す
  long long dfs(unsigned int v, unsigned int t, long long f) {
    if (v == t) {
      return f;
    }
    for (int &i = iter.at(v); i < graph.at(v).size(); ++i) {
      Edge &e = graph.at(v).at(i);
      // 流せてかつ最短経路が長くなるようなやつのみ
      if (e.cap - e.flow > 0 and level.at(v) < level.at(e.to)) {
        long long d = dfs(e.to, t, min(f, e.cap - e.flow));
        if (d > 0) {
          e.flow += d;
          graph.at(e.to).at(e.rev).flow -= d;
          return d;
        }
      }
    }
    return 0;
  }
};

int main() {
  ios::sync_with_stdio(0);
  cin.tie(0);

  int V, E;
  cin >> V >> E;
  Dinic dinic(V);
  for (int i = 0; i < E; i++) {
    int u, v, c;
    cin >> u >> v >> c;
    dinic.add_edge(u, v, c);
  }

  cout << dinic.max_flow(0, V - 1) << endl;
  return 0;
}


Error: TLE

Fixed submission:

#include <bits/stdc++.h>

#define FOR(i, a, b) for (int i = (a); i < ((int)b); ++i)
#define RFOR(i, a) for (int i = (a); i >= 0; --i)
#define FOE(i, a) for (auto i : a)
#define ALL(c) (c).begin(), (c).end()
#define RALL(c) (c).rbegin(), (c).rend()
#define DUMP(x) cerr << #x << " = " << (x) << endl;
#define SUM(x) std::accumulate(ALL(x), 0LL)
#define MIN(v) *std::min_element(v.begin(), v.end())
#define MAX(v) *std::max_element(v.begin(), v.end())
#define EXIST(v, x) (std::find(v.begin(), v.end(), x) != v.end())
#define BIT(n) (1LL << (n))
#define UNIQUE(v) v.erase(unique(v.begin(), v.end()), v.end());

typedef long long LL;
template <typename T> using V = std::vector<T>;
template <typename T> using VV = std::vector<std::vector<T>>;
template <typename T> using VVV = std::vector<std::vector<std::vector<T>>>;
template <class T> inline T ceil(T a, T b) { return (a + b - 1) / b; }
template <class T> inline void print(T x) { std::cout << x << std::endl; }
template <class T> inline bool inside(T y, T x, T H, T W) {
  return 0 <= y and y < H and 0 <= x and x < W;
}
inline double distance(double y1, double x1, double y2, double x2) {
  return sqrt((x1 - x2) * (x1 - x2) + (y1 - y2) * (y1 - y2));
}

const int INF = 1L << 30;
const double EPS = 1e-9;
const std::string YES = "YES", Yes = "Yes", NO = "NO", No = "No";
const std::vector<int> dy = {0, 1, 0, -1},
                       dx = {1, 0, -1, 0}; // 4近傍（右, 下, 左, 上）

using namespace std;

// 最大流問題を解く O(|E||V|^2)
class Dinic {

  struct Edge {
    const unsigned int to;  // 行き先
    long long flow;         // 流量
    const long long cap;    // 容量
    const unsigned int rev; // 逆辺
    const bool is_rev;      // 逆辺かどうか

    Edge(unsigned int to, long long flow, long long cap, unsigned int rev,
         bool is_rev)
        : to(to), flow(flow), cap(cap), rev(rev), is_rev(is_rev) {
      assert(this->cap >= 0);
    }
  };

  vector<vector<Edge>> graph; // グラフの隣接リスト表現
  vector<int> level;          // sからの距離
  vector<int> iter;           // dfsがどこまで進んだか

public:
  Dinic(unsigned int num_of_node) {
    assert(num_of_node > 0);
    graph.resize(num_of_node);
    level.resize(num_of_node);
    iter.resize(num_of_node);
  }

  // fromからtoへ向かう容量capの辺をグラフに追加する
  void add_edge(unsigned int from, unsigned int to, long long cap) {
    graph[from].push_back(
        Edge(to, 0, cap, (unsigned int)graph[to].size(), false));
    graph[to].push_back(
        Edge(from, cap, cap, (unsigned int)graph[from].size() - 1, true));
  }

  // sからtへの最大流を求める
  long long max_flow(unsigned int s, unsigned int t) {
    long long flow = 0;
    while (true) {
      // 残余ネットワーク上でsから各ノードへの距離を求める
      bfs(s);
      if (level.at(t) < 0) {
        return flow;
      }
      // 距離が増加する向きの辺のみを使って流せるだけ流す
      fill(iter.begin(), iter.end(), 0);
      long long f;
      while ((f = dfs(s, t, INT_MAX)) > 0) {
        flow += f;
      }
    }
  }

private:
  // 残余ネットワーク上でsから全ノードへの最短距離をBFSで計算する
  void bfs(unsigned int s) {
    fill(level.begin(), level.end(), -1);
    queue<unsigned int> que;
    level.at(s) = 0;
    que.push(s);
    while (not que.empty()) {
      unsigned int v = que.front();
      que.pop();
      for (int i = 0; i < graph.at(v).size(); ++i) {
        Edge &e = graph.at(v).at(i);
        if (e.cap - e.flow > 0 && level.at(e.to) < 0) {
          level.at(e.to) = level.at(v) + 1;
          que.push(e.to);
        }
      }
    }
  }

  // vからtへf流す
  long long dfs(unsigned int v, unsigned int t, long long f) {
    if (v == t) {
      return f;
    }
    for (int &i = iter.at(v); i < graph.at(v).size(); ++i) {
      Edge &e = graph.at(v).at(i);
      // 流せてかつ最短経路が長くなるようなやつのみ
      if (e.cap > 0 and level.at(v) < level.at(e.to)) {
        long long d = dfs(e.to, t, min(f, e.cap - e.flow));
        if (d > 0) {
          e.flow += d;
          graph.at(e.to).at(e.rev).flow -= d;
          return d;
        }
      }
    }
    return 0;
  }
};

int main() {
  ios::sync_with_stdio(0);
  cin.tie(0);

  int V, E;
  cin >> V >> E;
  Dinic dinic(V);
  for (int i = 0; i < E; i++) {
    int u, v, c;
    cin >> u >> v >> c;
    dinic.add_edge(u, v, c);
  }

  cout << dinic.max_flow(0, V - 1) << endl;
  return 0;
}