結果

問題 No.3111 Toll Optimization
コンテスト
ユーザー Hydrogen332
提出日時 2025-04-18 23:51:38
言語 C++23
(gcc 13.3.0 + boost 1.89.0)
結果
AC  
実行時間 171 ms / 5,000 ms
コード長 1,659 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 3,821 ms
コンパイル使用メモリ 290,208 KB
実行使用メモリ 14,908 KB
最終ジャッジ日時 2025-04-18 23:51:50
合計ジャッジ時間 11,330 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 70
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define rep(i, s, e) for (int i = (int)(s); i < (int)(e); ++i)
#define all(a) (a).begin(),(a).end()

const ll INF = 1ll << 60;

template<typename T> bool chmin(T& a, const T& b) {
    if (a > b) {
        a = b;
        return true;
    }
    else return false;
}

int main() {
    cin.tie(nullptr);
    ios_base::sync_with_stdio(false);
    
    int N, M, K;
    cin >> N >> M >> K;
    vector<ll> C(M);
    rep(i, 0, M) cin >> C[i];
    vector<vector<pair<int, int>>> G(N);
    rep(i, 0, M) {
        int u, v;
        cin >> u >> v;
        --u, --v;
        G[u].push_back({v, i});
        G[v].push_back({u, i});
    }
    
    vector dist(K + 1, vector<ll>(N, INF));
    priority_queue<pair<ll, pair<int, int>>, vector<pair<ll, pair<int, int>>>, greater<pair<ll, pair<int, int>>>> que;
    dist[K][0] = 0;
    que.push({0, {K, 0}});
    
    while (!que.empty()) {
        ll d = que.top().first;
        int k = que.top().second.first;
        int v = que.top().second.second;
        que.pop();
        
        if (d > dist[k][v]) continue;
        
        for (pair<int, int> e : G[v]) {
            if (k != 0) {
                if (chmin(dist[k - 1][e.first], dist[k][v])) {
                    que.push({dist[k - 1][e.first], {k - 1, e.first}});
                }
            }
            
            if (chmin(dist[k][e.first], dist[k][v] + C[e.second])) {
                que.push({dist[k][e.first], {k, e.first}});
            }
        }
    }
    
    ll ans = INF;
    rep(i, 0, K + 1) ans = min(ans, dist[i][N - 1]);
    
    cout << (ans == INF ? -1 : ans) << '\n';
}
0