結果
問題 |
No.1812 Uribo Road
|
ユーザー |
![]() |
提出日時 | 2022-01-14 22:17:31 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,185 ms / 5,000 ms |
コード長 | 1,953 bytes |
コンパイル時間 | 4,660 ms |
コンパイル使用メモリ | 263,160 KB |
最終ジャッジ日時 | 2025-01-27 11:41:57 |
ジャッジサーバーID (参考情報) |
judge3 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 30 |
ソースコード
#include <stdio.h> #include <bits/stdc++.h> #include <atcoder/all> using namespace atcoder; using mint = modint1000000007; using namespace std; #define rep(i,n) for(int i=0;i<(n);i++) #define Inf 2000000001 vector<int> a,b,c; vector<int> get(vector<vector<int>> &E,int s){ int N = E.size(); vector<int> dis(N,Inf); dis[s] = 0; priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>> Q; Q.emplace(0,s); while(Q.size()>0){ int D = Q.top().first; int u = Q.top().second; Q.pop(); rep(i,E[u].size()){ int ind = E[u][i]; int v = u ^ a[ind] ^ b[ind]; int DD = D + c[ind]; if(dis[v] > DD){ dis[v] = DD; Q.emplace(DD,v); } } } return dis; } int main(){ int N,M,K; cin>>N>>M>>K; vector<int> R(K); rep(i,K){ cin>>R[i]; R[i]--; } a.resize(M),b.resize(M),c.resize(M); vector<vector<int>> E(N); rep(i,M){ cin>>a[i]>>b[i]>>c[i]; a[i]--;b[i]--; E[a[i]].push_back(i); E[b[i]].push_back(i); } vector<int> t; t.push_back(0); t.push_back(N-1); rep(i,K){ t.push_back(a[R[i]]); t.push_back(b[R[i]]); } sort(t.begin(),t.end()); t.erase(unique(t.begin(),t.end()),t.end()); vector<vector<int>> ds(t.size()); rep(i,t.size())ds[i] = get(E,t[i]); vector dp(t.size(),vector<int>(1<<K,Inf)); dp[0][0] = 0; rep(i,3*K){ vector ndp(t.size(),vector<int>(1<<K,Inf)); rep(j,t.size()){ rep(k,1<<K){ if(dp[j][k]==Inf)continue; rep(l,t.size()){ ndp[l][k] = min(ndp[l][k],dp[j][k] + ds[j][t[l]]); } rep(l,K){ if(a[R[l]]==t[j]){ int to = distance(t.begin(),lower_bound(t.begin(),t.end(),b[R[l]])); ndp[to][k|(1<<l)] = min(ndp[to][k|(1<<l)], dp[j][k] + c[R[l]]); } if(b[R[l]]==t[j]){ int to = distance(t.begin(),lower_bound(t.begin(),t.end(),a[R[l]])); ndp[to][k|(1<<l)] = min(ndp[to][k|(1<<l)], dp[j][k] + c[R[l]]); } } } } swap(dp,ndp); } cout<<dp.back().back()<<endl; return 0; }