結果

問題 No.3111 Toll Optimization
コンテスト
ユーザー 0214sh7
提出日時 2025-04-19 19:05:05
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 455 ms / 5,000 ms
コード長 3,067 bytes
コンパイル時間 2,329 ms
コンパイル使用メモリ 210,064 KB
実行使用メモリ 112,016 KB
最終ジャッジ日時 2025-04-19 19:05:24
合計ジャッジ時間 17,072 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 70
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
typedef pair<ll,ll> PP;
// #define MOD 1000000007
#define MOD 998244353
#define INF 2305843009213693951
#define PI 3.141592653589
#define setdouble setprecision
#define REP(i,n) for(ll i=0;i<(n);++i)
#define OREP(i,n) for(ll i=1;i<=(n);++i)
#define RREP(i,n) for(ll i=(n)-1;i>=0;--i)
#define ORREP(i,n) for(ll i=(n);i>=1;--i)
#define rep(i,a,b) for(ll i=(a);i<=(b);++i)
#define ALL(v) (v).begin(), (v).end()
#define chmin(k,m) k = min(k,m)
#define chmax(k,m) k = max(k,m)
#define GOODBYE do { cout << "-1" << endl; return 0; } while (false)
#define MM <<" "<<
#define Endl endl

class Dijkstra{
    // Copyright (c) 2023 0214sh7
    // https://github.com/0214sh7/library/
    private:
    typedef std::pair<long long,int> P;
    std::vector<std::vector<P>> G;
    int V;
    // long long INF = (1LL<<61);
    std::priority_queue<P,std::vector<P>,std::greater<P>> que;
    
    public:
    void init(int N,std::vector<std::pair<std::pair<int,int>,long long>> edge){
        //頂点数を決定する
        V=N;
        
        //辺集合を扱いやすい形式に変換する
        G.resize(V);
        for(int i=0;i<edge.size();i++){
            int from=edge[i].first.first,to=edge[i].first.second;
            long long cost=edge[i].second;
            G[from].push_back({cost,to});
        }
    }
    
    Dijkstra(int N,std::vector<std::pair<std::pair<int,int>,long long>> edge){
        init(N,edge);
    }

    std::vector<long long> solve(int s){
        std::vector<long long> d;
        //INFで初期化する
        for(int i=0;i<V;i++){
            d.push_back(INF);
        }
        d[s]=0;
        que.push({0,s});
        //queは{cost,to}をコストが小さい順に出す
        while(!que.empty()){
            P p = que.top();
            que.pop();
            int v=p.second;
            if(d[v]<p.first)continue;
            for(int i=0;i<G[v].size();i++){
                P e = G[v][i];
                if(d[e.second]>d[v]+e.first){
                    d[e.second] = d[v]+e.first;
                    que.push({d[e.second],e.second});
                }
            }
        }
        return d;
    }
};

int main(void){
    //cin.tie(nullptr);
    //ios::sync_with_stdio(false);

    ll N,M,K;
    cin >> N >> M >> K;
    vector<ll> C(M);
    REP(i,M){cin >> C[i];}
    std::vector<std::pair<std::pair<int,int>,long long>> E;
    REP(i,M){
        ll u,v;
        cin >> u >> v;
        u--;v--;
        REP(k,K+1){
            // 使わない
            E.push_back({{u+k*N,v+k*N},C[i]});
            E.push_back({{v+k*N,u+k*N},C[i]});
            // 使う
            if(k==K)continue;
            E.push_back({{u+k*N,v+(k+1)*N},0});
            E.push_back({{v+k*N,u+(k+1)*N},0});
        }
    }
    
    Dijkstra dijkstra((K+1)*N,E);
    std::vector<long long> D = dijkstra.solve(0);

    ll Ans = INF;
    REP(k,K+1){
        chmin(Ans,D[(k+1)*N-1]);
    }
    if(Ans==INF)Ans=-1;
    cout << Ans << endl;


    return 0;
    
}
0