結果

問題 No.1812 Uribo Road
ユーザー 👑 Nachia
提出日時 2022-01-14 22:16:42
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 322 ms / 5,000 ms
コード長 2,066 bytes
コンパイル時間 1,421 ms
コンパイル使用メモリ 97,088 KB
最終ジャッジ日時 2025-01-27 11:40:18
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff # 


#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;
using i64 = long long;
using u64 = unsigned long long;
using i32 = int;
using u32 = unsigned int;
#define rep(i,n) for(int i=0; i<(n); i++)

template<class E>
using nega_queue = priority_queue<E,vector<E>,greater<E>>;

struct Edge{ int u,v,c; };
struct AdjEdge{ int to; i64 d; };

int N,M;
vector<Edge> J;
vector<vector<AdjEdge>> E;
int K;
vector<int> R;


vector<i64> dist(int s){
    nega_queue<pair<i64,int>> Q;
    vector<i64> D(E.size(), 1001001001001001001);
    Q.push({ 0,s });
    D[s] = 0;
    while(Q.size()){
        auto [d,p] = Q.top(); Q.pop();
        if(D[p] != d) continue;
        for(auto [to,dd] : E[p]){
            if(D[to] <= d+dd) continue;
            D[to] = d+dd;
            Q.push({ D[to],to });
        }
    }
    return D;
}


int main() {
    cin >> N >> M >> K;
    E.resize(N);
    J.resize(M);
    R.resize(K);
    rep(i,K){ cin >> R[i]; R[i]--; }
    rep(i,M){
        int u,v,c; cin >> u >> v >> c; u--; v--;
        J[i] = {u,v,c};
        E[u].push_back({v,c});
        E[v].push_back({u,c});
    }
    
    vector<int> Rp(K*2);
    rep(i,K) Rp[i*2] = J[R[i]].u;
    rep(i,K) Rp[i*2+1] = J[R[i]].v;

    vector<vector<AdjEdge>> G2(K*2 * (1<<K) + 2);
    const int ZG2 = K*2 * (1<<K);
    rep(i,K*2){
        auto D = dist(Rp[i]);
        rep(j,K*2){
            i64 d = D[Rp[j]];
            rep(b,1<<K) G2[(i<<K)+b].push_back({ (j<<K)+b, d });
        }
    }
    
    rep(k,K*2) rep(b,1<<K) G2[(k<<K)+b].push_back({ ((k^1)<<K)+(b|(1<<(k/2))), J[R[k/2]].c });

    {
        auto D = dist(0);
        rep(j,K*2) G2[ZG2].push_back({ j<<K, D[Rp[j]] });
    }
    {
        auto D = dist(N-1);
        rep(j,K*2) G2[((j+1)<<K)-1].push_back({ ZG2+1, D[Rp[j]] });
    }

    E = G2;
    auto D = dist(ZG2);
    i64 ans = D[ZG2+1];
    cout << ans << endl;

    return 0;
}



struct ios_do_not_sync{
    ios_do_not_sync(){
        std::ios::sync_with_stdio(false);
        std::cin.tie(nullptr);
    }
} ios_do_not_sync_instance;
0