結果

問題 No.654 Air E869120
コンテスト
ユーザー yuppe19 😺
提出日時 2019-03-21 15:02:55
言語 C++17
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 91 ms / 2,000 ms
コード長 2,075 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,259 ms
コンパイル使用メモリ 103,244 KB
最終ジャッジ日時 2025-01-06 23:38:52
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 35
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:48:21: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   48 |   int N, M, d; scanf("%d%d%d", &N, &M, &d);
      |                ~~~~~^~~~~~~~~~~~~~~~~~~~~~
main.cpp:54:29: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   54 |     int u, v, p, q, w; scanf("%d%d%d%d%d", &u, &v, &p, &q, &w);
      |                        ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff #
raw source code 

#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