結果

問題 No.614 壊れたキャンパス
ユーザー ei1333333
提出日時 2017-12-14 03:00:16
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 323 ms / 2,000 ms
コード長 1,641 bytes
コンパイル時間 2,506 ms
コンパイル使用メモリ 207,036 KB
最終ジャッジ日時 2025-01-05 05:18:59
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

using int64 = long long;
const int64 INF = 1LL << 60;

int main()
{
  int N, M, K, S, T;
  cin >> N >> M >> K >> S >> T;
  vector< pair< int, int > > ev[200000];
  for(int i = 0; i < M; i++) {
    int a, b, c;
    cin >> a >> b >> c;
    ev[a].emplace_back(c, b);
  }
  ev[0].emplace_back(S, -1);
  ev[N - 1].emplace_back(T, -1);
  for(int i = 0; i < N; i++) {
    sort(begin(ev[i]), end(ev[i]));
  }

  auto sweep = [&](vector< int64 > &dp, int idx)
  {
    for(int i = 1; i < dp.size(); i++) {
      dp[i] = min(dp[i], dp[i - 1] + ev[idx][i].first - ev[idx][i - 1].first);
    }
    for(int i = (int) dp.size() - 2; i >= 0; i--) {
      dp[i] = min(dp[i], dp[i + 1] + ev[idx][i + 1].first - ev[idx][i].first);
    }
  };

  vector< int64 > dp(ev[0].size(), INF);
  dp[lower_bound(begin(ev[0]), end(ev[0]), make_pair(S, -1)) - begin(ev[0])] = 0;
  sweep(dp, 0);


  for(int i = 1; i < N; i++) {
    vector< int64 > dp2(ev[i].size(), INF);
    for(int j = 0; j < ev[i].size(); j++) {
      auto &p = ev[i][j];
      if(p.second == -1) continue;
      int q = lower_bound(begin(ev[i - 1]), end(ev[i - 1]), make_pair(p.second, -1)) - begin(ev[i - 1]);

      if(q < ev[i - 1].size()) {
        dp2[j] = min(dp2[j], dp[q] + ev[i - 1][q].first - p.second);
      }
      --q;
      if(0 <= q && q < ev[i - 1].size()) {
        dp2[j] = min(dp2[j], dp[q] + p.second - ev[i - 1][q].first);
      }
    }
    sweep(dp2, i);
    dp.swap(dp2);
  }

  auto ret = dp[lower_bound(begin(ev[N - 1]), end(ev[N - 1]), make_pair(T, -1)) - begin(ev[N - 1])];
  if(ret >= INF) ret = -1;
  cout << ret << endl;

}
0