結果

問題 No.1814 Uribo Road (Easy)
ユーザー 👑 potato167
提出日時 2022-01-15 20:00:50
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 10 ms / 2,000 ms
コード長 2,813 bytes
コンパイル時間 2,299 ms
コンパイル使用メモリ 213,456 KB
最終ジャッジ日時 2025-01-27 12:37:42
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 44
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#pragma GCC optimize("Ofast")
#define _GLIBCXX_DEBUG
using namespace std;
using std::cout;
using std::cin;
using std::endl;
using ll=long long;
using ld=long double;
ll ILL=1167167167167167167;
const int INF=2100000000;
const ll mod=1e9+7;
#define rep(i,a) for (ll i=0;i<a;i++)
template<class T> using _pq = priority_queue<T, vector<T>, greater<T>>;
template<class T> ll LB(vector<T> &v,T a){return lower_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> ll UB(vector<T> &v,T a){return upper_bound(v.begin(),v.end(),a)-v.begin();}
template<class T> bool chmin(T &a,const T &b){if(a>b){a=b;return 1;}else return 0;}
template<class T> bool chmax(T &a,const T &b){if(a<b){a=b;return 1;}else return 0;}
template<class T> void So(vector<T> &v) {sort(v.begin(),v.end());}
template<class T> void Sore(vector<T> &v) {sort(v.begin(),v.end(),[](T x,T y){return x>y;});}
void yneos(bool a){if(a) cout<<"Yes\n"; else cout<<"No\n";}


using len_type=int;
struct edge{
    int to;
    len_type length;
};

len_type dijkstra_movement(edge e,int from,len_type t){
    return t+e.length;
}


vector<len_type> dijkstra(vector<vector<edge>> &G,int start,len_type inf){
    int M=G.size();
    assert(M>start&&start>=0);
    vector<len_type> seen(M,inf);
    seen[start]=0;
    priority_queue<pair<len_type,int>,vector<pair<len_type,int>>,greater<pair<len_type,int>>> pq;
    pq.push({0,start});
    while(!pq.empty()){
        len_type time;
        int ver;
        tie(time,ver)=pq.top();
        pq.pop();
        if(seen[ver]<time) continue;
        for(edge x:G[ver]){
            len_type T=dijkstra_movement(x,ver,time);
            if(T<seen[x.to]){
                pq.push({T,x.to});
                seen[x.to]=T;
            }
        }
    }
    return seen;
}

//  rainy ~ 雨に打たれて ~
int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	
	int N,M,K;
	cin>>N>>M>>K;
	vector<int> R(K);
	rep(i,K) cin>>R[i],R[i]--;
	vector<int> A(M),B(M),C(M);
	vector<vector<edge>> G(N);
	rep(i,M){
		cin>>A[i]>>B[i]>>C[i];
		A[i]--,B[i]--;
		G[A[i]].push_back({B[i],C[i]});
		G[B[i]].push_back({A[i],C[i]});
	}
	vector<vector<int>> seen(K*2);
	rep(i,K){
		seen[i*2]=dijkstra(G,A[R[i]],INF);
		seen[i*2+1]=dijkstra(G,B[R[i]],INF);
	}
	vector<vector<int>> dp((1<<K),vector<int>(K*2,INF));
	rep(i,K){
		dp[(1<<i)][i*2]=seen[i*2+1][0]+C[R[i]];
		dp[(1<<i)][i*2+1]=seen[i*2][0]+C[R[i]];
	}
	rep(i,(1<<K)){
		rep(j,K*2){
			if(dp[i][j]==INF) continue;
			rep(k,K){
				if(i&(1<<k)) continue;
				int tmp=i+(1<<k);
				chmin(dp[tmp][k*2],dp[i][j]+seen[j][B[R[k]]]+C[R[k]]);
				chmin(dp[tmp][k*2+1],dp[i][j]+seen[j][A[R[k]]]+C[R[k]]);
			}
		}
	}
	int ans=INF;
	rep(j,K*2){
		//cout<<dp[(1<<K)-1][j]<<" "<<seen[j][N-1]<<"\n";
		chmin(ans,dp[(1<<K)-1][j]+seen[j][N-1]);
	}
	cout<<ans<<"\n";
}
0