結果

問題 No.654 Air E869120
ユーザー yuppe19 😺yuppe19 😺
提出日時 2019-03-21 15:02:55
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 99 ms / 2,000 ms
コード長 2,075 bytes
コンパイル時間 1,171 ms
コンパイル使用メモリ 109,576 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-09-19 01:38:23
合計ジャッジ時間 3,636 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 1 ms
6,944 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 2 ms
6,940 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 AC 2 ms
6,944 KB
testcase_09 AC 1 ms
6,944 KB
testcase_10 AC 47 ms
6,944 KB
testcase_11 AC 34 ms
6,940 KB
testcase_12 AC 36 ms
6,940 KB
testcase_13 AC 43 ms
6,944 KB
testcase_14 AC 17 ms
6,944 KB
testcase_15 AC 18 ms
6,944 KB
testcase_16 AC 65 ms
6,944 KB
testcase_17 AC 99 ms
6,940 KB
testcase_18 AC 63 ms
6,944 KB
testcase_19 AC 53 ms
6,940 KB
testcase_20 AC 51 ms
6,940 KB
testcase_21 AC 53 ms
6,940 KB
testcase_22 AC 51 ms
6,940 KB
testcase_23 AC 51 ms
6,940 KB
testcase_24 AC 51 ms
6,940 KB
testcase_25 AC 49 ms
6,940 KB
testcase_26 AC 49 ms
6,940 KB
testcase_27 AC 49 ms
6,944 KB
testcase_28 AC 48 ms
6,940 KB
testcase_29 AC 48 ms
6,940 KB
testcase_30 AC 52 ms
6,944 KB
testcase_31 AC 52 ms
6,940 KB
testcase_32 AC 51 ms
6,944 KB
testcase_33 AC 52 ms
6,944 KB
testcase_34 AC 52 ms
6,944 KB
testcase_35 AC 2 ms
6,944 KB
testcase_36 AC 1 ms
6,940 KB
testcase_37 AC 1 ms
6,944 KB
testcase_38 AC 1 ms
6,944 KB
testcase_39 AC 2 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <map>
#include <vector>
using namespace std;
using i64 = int64_t;

struct edge { int to; i64 avail; int rev; };
constexpr i64 inf = 987'654'321'987'654'321LL;

int n;
vector<vector<edge>> G;
vector<int> ptr;
vector<bool> used;

void add_edge(int u, int v, i64 c) {
  G[u].push_back(edge{v, c, ptr[v]++});
  G[v].push_back(edge{u, 0, ptr[u]++});
}

i64 dfs(int v, int t, i64 flow) {
  if(v == t) { return flow; }
  used[v] = true;
  for(edge &e : G[v]) {
    edge &x = G[e.to][e.rev];
    if(!used[e.to] && e.avail) {
      i64 res = dfs(e.to, t, min(flow, e.avail));
      if(res) {
        e.avail -= res;
        x.avail += res;
        return res;
      }
    }
  }
  return 0;
}

i64 max_flow(int s, int t) {
  i64 res = 0;
  for(;;) {
    used.assign(n, false);
    i64 flow = dfs(s, t, inf);
    if(flow == 0) { return res; }
    res += flow;
  }
}

int main(void) {
  int N, M, d; scanf("%d%d%d", &N, &M, &d);
  vector<tuple<int, int, int, int, int>> logs;
  // mp[(街, 時刻, 発か)] := id
  map<tuple<int, int, bool>, int> mp;
  int idx = 0;
  for(int i=0; i<M; ++i) {
    int u, v, p, q, w; scanf("%d%d%d%d%d", &u, &v, &p, &q, &w);
    logs.emplace_back(u, v, p, q, w);
    auto k0 = make_tuple(u, p, true),
         k1 = make_tuple(v, q, false);
    if(!mp.count(k0)) { mp[k0] = idx++; }
    if(!mp.count(k1)) { mp[k1] = idx++; }
  }
  ::n = idx + 2;
  int src = idx, dst = idx + 1;
  G.assign(n, vector<edge>());
  ptr.assign(n, 0);

  for(auto [u, v, p, q, w] : logs) {
    auto k0 = make_tuple(u, p, true),
         k1 = make_tuple(v, q, false);
    add_edge(mp[k0], mp[k1], w);
  }

  for(auto&& [key0, _] : mp) {
    auto&& [u0, t0, b0] = key0;

    if(u0 == 1 && b0) {
      add_edge(src, mp[key0], inf);
    }
    if(u0 == N && !b0) {
      add_edge(mp[key0], dst, inf);
    }

    for(auto&& [key1, __] : mp) {
      auto [u1, t1, b1] = key1;
      if(!b0 && b1 && u0 == u1 && t0+d <= t1) {
        add_edge(mp[key0], mp[key1], inf);
      }
    }
  }
  i64 maxi = max_flow(src, dst);
  printf("%ld\n", maxi);
  return 0;
}
0