結果

問題 No.1814 Uribo Road (Easy)
ユーザー 👑 NachiaNachia
提出日時 2022-01-15 20:14:32
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 10 ms / 2,000 ms
コード長 2,065 bytes
コンパイル時間 1,591 ms
コンパイル使用メモリ 100,708 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-05-01 18:26:32
合計ジャッジ時間 2,761 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,940 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 2 ms
6,940 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 2 ms
6,944 KB
testcase_09 AC 3 ms
6,940 KB
testcase_10 AC 3 ms
6,944 KB
testcase_11 AC 3 ms
6,940 KB
testcase_12 AC 3 ms
6,940 KB
testcase_13 AC 3 ms
6,940 KB
testcase_14 AC 4 ms
6,940 KB
testcase_15 AC 2 ms
6,944 KB
testcase_16 AC 3 ms
6,944 KB
testcase_17 AC 3 ms
6,940 KB
testcase_18 AC 2 ms
6,940 KB
testcase_19 AC 3 ms
6,944 KB
testcase_20 AC 5 ms
6,944 KB
testcase_21 AC 3 ms
6,940 KB
testcase_22 AC 9 ms
6,944 KB
testcase_23 AC 5 ms
6,944 KB
testcase_24 AC 8 ms
6,944 KB
testcase_25 AC 8 ms
6,940 KB
testcase_26 AC 9 ms
6,940 KB
testcase_27 AC 10 ms
6,944 KB
testcase_28 AC 10 ms
6,944 KB
testcase_29 AC 10 ms
6,940 KB
testcase_30 AC 9 ms
6,940 KB
testcase_31 AC 9 ms
6,940 KB
testcase_32 AC 4 ms
6,940 KB
testcase_33 AC 2 ms
6,944 KB
testcase_34 AC 2 ms
6,940 KB
testcase_35 AC 3 ms
6,940 KB
testcase_36 AC 3 ms
6,940 KB
testcase_37 AC 3 ms
6,944 KB
testcase_38 AC 4 ms
6,940 KB
testcase_39 AC 3 ms
6,940 KB
testcase_40 AC 3 ms
6,944 KB
testcase_41 AC 7 ms
6,944 KB
testcase_42 AC 5 ms
6,940 KB
testcase_43 AC 4 ms
6,944 KB
testcase_44 AC 5 ms
6,944 KB
testcase_45 AC 3 ms
6,940 KB
testcase_46 AC 7 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

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