結果

問題 No.3111 Toll Optimization
ユーザー tombo_
提出日時 2025-04-18 20:51:22
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 256 ms / 5,000 ms
コード長 2,822 bytes
コンパイル時間 1,114 ms
コンパイル使用メモリ 105,584 KB
実行使用メモリ 20,048 KB
最終ジャッジ日時 2025-04-18 20:51:31
合計ジャッジ時間 9,168 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 70
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<iostream>
#include<vector>
#include<algorithm>
#include<iomanip>
#include<queue>
#include<set>
#include<map>
using namespace std;
#define OVERLOAD_REP(_1, _2, _3, name, ...) name
#define REP1(i, n) for (auto i = std::decay_t<decltype(n)>{}; (i) != (n); ++(i))
#define REP2(i, l, r) for (auto i = (l); (i) != (r); ++(i))
#define rep(...) OVERLOAD_REP(__VA_ARGS__, REP2, REP1)(__VA_ARGS__)
#define REP(i, l, r) rep(i, l, r+1)
#define all(x) (x).begin(),(x).end()
#define rall(x) (x).rbegin(),(x).rend()
using ll = long long;
using ld = long double;
using P = pair<ll,ll>;
struct Edge {
    int to; ll w;
};
//using Graph = vector<vector<int> >;
using Graph = vector<vector<Edge> >;
const ll INF = 2e18;
//const int INF = 2e9;
template<class T> using vc = vector<T>;
template<class T> using vv = vector<vector<T> >;
template<class T> using vvv = vector<vector<vector<T> > >;
template<class T> using pq = priority_queue<T>;
template<class T> using pq_g = priority_queue<T, vc<T>, greater<T> >;
template<class T> istream& operator>>(istream& i, vc<T>& v) { rep(j, 0, v.size()) i>>v[j]; return i; }
template<class T> ostream& operator<<(ostream& o, vc<T>& v) { rep(j, 0, v.size()) o<<v[j]<<" "; return o; }
template<class T> bool chmin(T& a, T b) {
    if(a > b) {
        a = b;
        return true;
    }
    return false;
}
template<class T> bool chmax(T& a, T b) {
    if(a < b) {
        a = b;
        return true;
    }
    return false;
}

int main() {
    // 高速化
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    // 小数点の出力桁数を指定
    cout << fixed << setprecision(10);

    // メイン
    // ダイクストラ法
    // dist[v][n] -> n回チケットを使った時のvに来るまでの最小のコスト
    int N, M, K;
    cin >> N >> M >> K;
    vc<ll> C(M);
    cin >> C;
    Graph G(N);
    rep(i, 0, M) {
        int u, v;
        cin >> u >> v;
        u--, v--;
        G[u].push_back({v, C[i]});
        G[v].push_back({u, C[i]});
    }

    vv<ll> dist(N, vc<ll>(K+1, INF));
    dist[0][0] = 0;
    pq_g<pair<ll, P> > que; // 頂点、チケット数
    que.push({dist[0][0], {0, 0}});
    while(!que.empty()) {
        int v = que.top().second.first, t = que.top().second.second;
        ll c = que.top().first;
        que.pop();

        if(c > dist[v][t]) continue;
        for(Edge e : G[v]) {
            if(chmin(dist[e.to][t], c + e.w)) {
                que.push({dist[e.to][t], P(e.to, t)});
            }
            //cout << t << " " << c << " " << dist[e.to][t+1] << endl;
            if(t + 1 <= K && chmin(dist[e.to][t+1], c)) {
                que.push({dist[e.to][t+1], P(e.to, t+1)});
            }
        }
    }
    ll res = INF;
    REP(i, 0, K) chmin(res, dist[N-1][i]);
    cout << (res == INF ? -1 : res) << endl;


    return 0;
}
0