結果

問題 No.3111 Toll Optimization
ユーザー hirayuu_yc
提出日時 2025-04-18 20:43:13
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 306 ms / 5,000 ms
コード長 1,954 bytes
コンパイル時間 3,889 ms
コンパイル使用メモリ 295,096 KB
実行使用メモリ 17,708 KB
最終ジャッジ日時 2025-04-18 20:43:30
合計ジャッジ時間 14,261 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 70
権限があれば一括ダウンロードができます

ソースコード

diff # 

/*
from heapq import*
N,M,K=map(int,input().split())
gr=[[] for i in range(N)]
C=list(map(int,input().split()))
for i in range(M):
    u,v=map(int,input().split())
    gr[u-1].append((v-1,C[i]))
    gr[v-1].append((u-1,C[i]))
dist=[[1<<60]*(K+1) for i in range(N)]
dist[0][0]=0
hq=[(0,0,0)]
while hq:
    d,p,c=heappop(hq)
    for np,cs in gr[p]:
        if dist[np][c]>d+cs:
            dist[np][c]=d+cs
            heappush(hq,(d+cs,np,c))
        if c!=K and dist[np][c+1]>d:
            dist[np][c+1]=d
            heappush(hq,(d,np,c+1))
ans=min(dist[-1])
if ans==(1<<60):
    print(-1)
else:
    print(ans)
*/

#include <bits/stdc++.h>
using namespace std;

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

int main() {
    int N, M, K;
    cin >> N >> M >> K;

    vector<vector<pair<int, int>>> gr(N);
    vector<int> C(M);
    for (int i = 0; i < M; ++i) {
        cin >> C[i];
    }
    for (int i = 0; i < M; ++i) {
        int u, v;
        cin >> u >> v;
        --u; --v;
        gr[u].emplace_back(v, C[i]);
        gr[v].emplace_back(u, C[i]);
    }

    vector<vector<ll>> dist(N, vector<ll>(K + 1, INF));
    dist[0][0] = 0;

    // Min-heap: (distance, node, free edge count)
    priority_queue<tuple<ll, int, int>, vector<tuple<ll, int, int>>, greater<>> hq;
    hq.emplace(0, 0, 0);

    while (!hq.empty()) {
        auto [d, p, c] = hq.top(); hq.pop();
        if (dist[p][c] < d) continue; // Skip outdated entries

        for (auto [np, cs] : gr[p]) {
            if (dist[np][c] > d + cs) {
                dist[np][c] = d + cs;
                hq.emplace(d + cs, np, c);
            }
            if (c != K && dist[np][c + 1] > d) {
                dist[np][c + 1] = d;
                hq.emplace(d, np, c + 1);
            }
        }
    }

    ll ans = *min_element(dist[N - 1].begin(), dist[N - 1].end());
    if (ans == INF) {
        cout << -1 << '\n';
    } else {
        cout << ans << '\n';
    }

    return 0;
}
0