結果

問題 No.614 壊れたキャンパス
コンテスト
ユーザー yuppe19 😺
提出日時 2017-12-14 14:59:44
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,665 bytes
コンパイル時間 1,584 ms
コンパイル使用メモリ 117,016 KB
最終ジャッジ日時 2025-01-05 05:24:10
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 18 TLE * 2
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:20:8: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   20 |   scanf("%d%d%d%d%d", &N, &M, &K, &S, &T);
      |   ~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
main.cpp:29:10: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   29 |     scanf("%d%d%d", &a, &b, &c);
      |     ~~~~~^~~~~~~~~~~~~~~~~~~~~~

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <unordered_map>
#include <queue>
#include <unordered_set>
using namespace std;
using i64 = long long;

constexpr i64 inf = 987654321987654321LL;
int N, M, K, S, T;
int a, b, c;
i64 cc; int p, q;
i64 _cost;
i64 mini;
int k; i64 v;
int src, dst;
int dif;

int main(void) {
  scanf("%d%d%d%d%d", &N, &M, &K, &S, &T);
  --S, --T;
  priority_queue<tuple<i64, int, int>, vector<tuple<i64, int, int>>, greater<tuple<i64, int, int>>> pq;
  // コスト, 棟, 階
  pq.emplace(0, 0, S);
  vector<unordered_map<int, i64>> cost(N); // cost[棟][階] := 最小コスト
  cost[0][S] = 0;
  vector<vector<pair<int, int>>> G(N, vector<pair<int, int>>()); // G[i] := {(i棟の出次階, i+1棟の入次階)のリスト}
  for(int i=0; i<M; ++i) {
    scanf("%d%d%d", &a, &b, &c);
    --a, --b, --c;
    G[a].emplace_back(b, c);
  }
  vector<unordered_set<int>> seen(N);
  mini = inf;
  while(!pq.empty()) {
    // 今 p棟 q階にいる。
    tie(cc, p, q) = pq.top(); pq.pop();
    if(!cost[p].count(q)) { continue; }
    _cost = cost[p][q];
    if(_cost < cc) { continue; }
    if(p == N-1) {
      mini = min(mini, _cost + abs(T - q));
    }
    if(seen[p].count(q)) { continue; }
    seen[p].insert(q);
    // 同棟の src 階に移動し、
    // 渡り廊下で p+1棟 dst階に移動する
    for(auto&& pr : G[p]) {
      dif = abs(q - pr.first);
      if(!cost[p+1].count(pr.second) || cost[p+1][pr.second] > _cost + dif) {
        cost[p+1][pr.second] = _cost + dif;
        pq.emplace(cost[p+1][pr.second], p+1, pr.second);
      }
    }
  }
  if(mini == inf) { mini = -1; }
  printf("%lld\n", mini);
  return 0;
}
0