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

#include <bits/stdc++.h>
#define int long long
#define INF 1e9
#define MAX_V 10000
using namespace std;

typedef pair<int, int> P;

// 辺を表す構造体(行き先、容量、逆辺)
struct edge {
  int to, cap, rev, num;
};

vector<edge> G[MAX_V]; // グラフの隣接リスト
bool used[MAX_V];      // DFSですでに調べたかのフラグ

int t_idx[MAX_V];

// fromからtoへ向かう容量capの辺をグラフに追加する
void add_edge(int from, int to, int cap, int num) {
  t_idx[num] = to;
  G[from].push_back((edge){to, cap, (int)G[to].size(), num});
  G[to].push_back((edge){from, 0, (int)G[from].size() - 1, num});
}

// 増加パスをDFSで探す
int dfs(int v, int t, int f) {
  if (v == t)
    return f;
  used[v] = true;
  for (int i = 0; i < G[v].size(); i++) {
    edge &e = G[v][i];
    if (!used[e.to] && e.cap > 0) {
      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;
}

// sからtへの最大流を求める
int max_flow(int s, int t) {
  int flow = 0;
  for (;;) {
    memset(used, 0, sizeof(used));
    int f = dfs(s, t, INF);
    if (f == 0)
      return flow;
    flow += f;
  }
}

int N, M, S, T;
int used1[MAX_V], used2[MAX_V];

void dfs1(int x) {

  if (used1[x])
    return;
  used1[x] = 1;

  for (edge nx : G[x]) {

    if (t_idx[nx.num] == x && nx.cap == 1)
      dfs1(nx.to);

    if (t_idx[nx.num] == nx.to && nx.cap == 1)
      dfs1(nx.to);
  }
}

void dfs2(int x) {

  if (used2[x])
    return;
  used2[x] = 1;

  for (edge nx : G[x]) {

    if (t_idx[nx.num] == x && nx.cap == 0)
      dfs2(nx.to);

    if (t_idx[nx.num] == nx.to && nx.cap == 0)
      dfs2(nx.to);
  }
}

signed main() {

  while (1) {

    cin >> N >> M >> S >> T;
    if (!N && !M && !S && !T)
      break;

    S--, T--;

    for (int i = 0; i < N; i++)
      G[i].clear();

    for (int i = 0; i < M; i++) {
      int a, b;
      cin >> a >> b;
      add_edge(a - 1, b - 1, 1, i);
    }

    int ans = max_flow(S, T), cnt = 0;

    memset(used1, 0, sizeof(used1));
    memset(used2, 0, sizeof(used2));

    dfs1(S);

    dfs2(T);

    for (int i = 0; i < N; i++) {

      for (edge ni : G[i]) {

        if (used2[i] && used1[ni.to] && t_idx[ni.num] == ni.to && ni.cap == 1)
          cnt++;
      }
    }

    if (cnt)
      ans++;

    cout << ans << ' ' << cnt << endl;
  }

  return 0;
}


Error: TLE

Fixed submission:

#include <bits/stdc++.h>
#define MAX_V 1005
#define INF 1e9
using namespace std;

/*最大流(Dinic法) O(|E||V|^2)*/
// 辺を表す構造体(行き先、容量、逆辺)
struct edge {
  int to, cap, rev, s;
};
vector<edge> G[MAX_V]; // グラフの隣接リスト表現
int level[MAX_V];      // sからの距離
int iter[MAX_V];       // どこまで調べ終わったか
// 辺を表す構造体(行き先、容量、逆辺)

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

// sからの最短距離をBFSで計算する
void bfs(int s) {
  memset(level, -1, sizeof(level));
  queue<int> que;
  level[s] = 0;
  que.push(s);
  while (!que.empty()) {
    int v = que.front();
    que.pop();
    for (int i = 0; i < G[v].size(); i++) {
      edge &e = G[v][i];
      if (e.cap > 0 && level[e.to] < 0) {
        level[e.to] = level[v] + 1;
        que.push(e.to);
      }
    }
  }
}

// 増加パスをDFSで探す。
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;
}

// sからtへの最大流を求める
int max_flow(int s, int t, int maxf) {
  int flow = 0;
  for (;;) {
    bfs(s);
    if (maxf == 0 || level[t] < 0)
      return flow;
    memset(iter, 0, sizeof(iter));
    int f;
    while ((f = dfs(s, t, maxf)) > 0)
      flow += f, maxf -= f;
  }
}

int n, m, s, t;

void solve() {

  int r = max_flow(s, t, INF);
  int maxflow = r;

  unordered_map<int, int> cnt;

  for (int i = 1; i <= n; i++) {

    for (int j = 0; j < G[i].size(); j++) {

      edge &s1 = G[i][j];
      edge &t1 = G[s1.to][s1.rev];

      if (s1.s)
        continue;
      if (t1.cap == 1)
        continue;

      s1.cap = 0;

      int f1 = max_flow(s, s1.to, 1);

      int f2 = max_flow(i, t, 1);

      int f = r;

      if (f1 && f2)
        f = r + 1;

      if (f1)
        max_flow(s1.to, s, 1);

      if (f2)
        max_flow(t, i, 1);

      if (maxflow < f)
        maxflow = f;

      cnt[f]++;

      s1.cap = 1;
    }
  }

  int revcnt = 0;

  if (r < maxflow) {
    for (pair<int, int> d : cnt)
      if (maxflow == d.first)
        revcnt += d.second;
  }

  cout << maxflow << ' ' << revcnt << endl;
}

void init() {
  for (int i = 0; i < MAX_V; i++)
    G[i].clear();
}

int main() {

  while (1) {

    cin >> n >> m >> s >> t;
    if (!n && !m && !s && !t)
      break;

    init();

    for (int i = 0; i < m; i++) {
      int a, b;
      cin >> a >> b;
      add_edge(a, b, 1);
    }

    solve();
  }

  return 0;
}