結果

問題 No.3111 Toll Optimization
コンテスト
ユーザー GOTKAKO
提出日時 2025-04-18 20:26:38
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 350 ms / 5,000 ms
コード長 2,195 bytes
コンパイル時間 2,384 ms
コンパイル使用メモリ 208,676 KB
実行使用メモリ 49,992 KB
最終ジャッジ日時 2025-04-18 20:26:53
合計ジャッジ時間 12,754 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 70
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

vector<long long> dijkstra(vector<vector<pair<int,long long>>> &Graph,int start){
int N = Graph.size();
//O((V+E)logV) 一般最短経路魔法.
long long inf = 3e18;
    vector<bool> used(N);
    vector<long long> ret(N,inf);
    priority_queue<pair<long long,int>,vector<pair<long long,int>>,greater<pair<long long,int>>> Q;
    ret.at(start) = 0; Q.push({0,start});
    while(Q.size()){
        auto[nowd,pos] = Q.top(); Q.pop();
        if(used.at(pos)) continue;
        used.at(pos) = true;
        for(auto [to,w] : Graph.at(pos)){
            if(ret.at(to) > nowd+w){
                ret.at(to) = nowd+w;
                Q.push({ret.at(to),to});
            }
        }
    }
    return ret;
}
vector<long long> dijkstra2(vector<vector<pair<int,long long>>> &Graph,int start){
    int N = Graph.size();
    //O(V^2) 密グラフ専用.
    long long inf = 3e18;
    vector<bool> used(N);
    vector<long long> ret(N,inf);
    ret.at(start) = 0;
    while(true){
        long long nowd = inf,pos = -1;
        for(int i=0; i<N; i++){
            if(used.at(i)) continue;
            if(nowd > ret.at(i)) nowd = ret.at(i),pos = i;
        }
        if(pos == -1) break;
        used.at(pos) = true;
 
        for(auto [to,w] : Graph.at(pos)){if(ret.at(to) > nowd+w) ret.at(to) = nowd+w;}
    }
    return ret;
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int N,M,K; cin >> N >> M >> K;
    vector<long long> C(M);
    for(auto &c : C) cin >> c;

    vector<vector<pair<int,long long>>> Graph((K+1)*N);
    for(int i=0; i<M; i++){
        int u,v; cin >> u >> v;
        u--; v--;
        long long w = C.at(i);
        for(int k=0; k<=K; k++){
            Graph.at(N*k+u).push_back({N*k+v,w});
            Graph.at(N*k+v).push_back({N*k+u,w});
            if(k < K){
                Graph.at(N*k+v).push_back({N*k+u+N,0});
                Graph.at(N*k+u).push_back({N*k+v+N,0});
            }
        }
    }

    auto dist = dijkstra(Graph,0);
    long long answer = 1e18;
    for(int i=0; i<=K; i++) answer = min(answer,dist.at(N-1+N*i));
    if(answer > 1e17) answer = -1;
    cout << answer << endl;
}
0