結果
| 問題 |
No.1812 Uribo Road
|
| コンテスト | |
| ユーザー |
 AngrySadEight
|
| 提出日時 | 2025-08-25 01:02:51 |
| 言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
| 結果 |
AC
|
| 実行時間 | 180 ms / 5,000 ms |
| コード長 | 6,300 bytes |
| コンパイル時間 | 3,937 ms |
| コンパイル使用メモリ | 203,284 KB |
| 実行使用メモリ | 10,488 KB |
| 最終ジャッジ日時 | 2025-08-25 01:02:59 |
| 合計ジャッジ時間 | 7,201 ms |
|
ジャッジサーバーID （参考情報） |
judge3 / judge4 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 30 |
ソースコード
#include <algorithm>
#include <atcoder/all>
#include <bitset>
#include <cassert>
#include <cmath>
#include <ctime>
#include <functional>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <random>
#include <set>
#include <stack>
#include <string>
#include <tuple>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
using namespace std;
using namespace atcoder;
typedef long long ll;
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
#define repr(i, n) for (int i = (int)(n) - 1; i >= 0; i--)
#define repk(i, k, n) for (int i = k; i < (int)(n); i++)
#define all(v) v.begin(), v.end()
#define mod1 1000000007
#define mod2 998244353
#define mod3 100000007
#define vi vector<int>
#define vs vector<string>
#define vc vector<char>
#define vl vector<ll>
#define vb vector<bool>
#define vvi vector<vector<int>>
#define vvc vector<vector<char>>
#define vvl vector<vector<ll>>
#define vvb vector<vector<bool>>
#define vvvi vector<vector<vector<int>>>
#define vvvl vector<vector<vector<ll>>>
#define pii pair<int, int>
#define pil pair<int, ll>
#define pli pair<ll, int>
#define pll pair<ll, ll>
#define vpii vector<pair<int, int>>
#define vpll vector<pair<ll, ll>>
#define vvpii vector<vector<pair<int, int>>>
#define vvpll vector<vector<pair<ll, ll>>>
// using mint = modint998244353;
template <typename T>
void debug(T e) {
cerr << e << endl;
}
template <typename T>
void debug(vector<T> &v) {
rep(i, v.size()) { cerr << v[i] << " "; }
cerr << endl;
}
template <typename T>
void debug(vector<vector<T>> &v) {
rep(i, v.size()) {
rep(j, v[i].size()) { cerr << v[i][j] << " "; }
cerr << endl;
}
}
template <typename T>
void debug(vector<pair<T, T>> &v) {
rep(i, v.size()) { cerr << v[i].first << " " << v[i].second << endl; }
}
template <typename T>
void debug(set<T> &st) {
for (auto itr = st.begin(); itr != st.end(); itr++) {
cerr << *itr << " ";
}
cerr << endl;
}
template <typename T>
void debug(multiset<T> &ms) {
for (auto itr = ms.begin(); itr != ms.end(); itr++) {
cerr << *itr << " ";
}
cerr << endl;
}
template <typename T>
void debug(map<T, T> &mp) {
for (auto itr = mp.begin(); itr != mp.end(); itr++) {
cerr << itr->first << " " << itr->second << endl;
}
}
void debug_out() { cerr << endl; }
template <typename Head, typename... Tail>
void debug_out(Head H, Tail... T) {
cerr << H << " ";
debug_out(T...);
}
using mint = modint998244353;
void debug_mint1(vector<mint> &vec) {
for (int i = 0; i < vec.size(); i++) {
cerr << vec[i].val() << " ";
}
cerr << endl;
}
void debug_mint2(vector<vector<mint>> &vec) {
for (int i = 0; i < vec.size(); i++) {
for (int j = 0; j < vec[i].size(); j++) {
cerr << vec[i][j].val() << " ";
}
cerr << endl;
}
}
void dijkstra(vvpll &graph, ll v, vl &dist) {
priority_queue<pll, vpll, greater<pll>> pque;
pque.push(pll(0, v));
dist[v] = 0;
while (pque.size()) {
pll p = pque.top();
pque.pop();
if (dist[p.second] != p.first) {
continue;
}
for (auto x : graph[p.second]) {
if (dist[x.first] > dist[p.second] + x.second) {
dist[x.first] = dist[p.second] + x.second;
pque.push(pll(dist[x.first], x.first));
}
}
}
}
int main() {
ll N, M, K;
cin >> N >> M >> K;
vector<ll> R(K);
for (ll i = 0; i < K; i++) {
cin >> R[i];
R[i]--;
}
vector<ll> A(M);
vector<ll> B(M);
vector<ll> C(M);
for (ll i = 0; i < M; i++) {
cin >> A[i] >> B[i] >> C[i];
A[i]--;
B[i]--;
}
vector<vector<pll>> graph(N, vector<pll>(0));
for (ll i = 0; i < M; i++) {
graph[A[i]].push_back(make_pair(B[i], i));
graph[B[i]].push_back(make_pair(A[i], i));
}
vector<vector<pll>> graph2(N, vector<pll>(0));
for (ll i = 0; i < M; i++) {
graph2[A[i]].push_back(make_pair(B[i], C[i]));
graph2[B[i]].push_back(make_pair(A[i], C[i]));
}
map<ll, ll> mp;
mp[0]++;
mp[N - 1]++;
for (ll i = 0; i < K; i++) {
mp[A[R[i]]]++;
mp[B[R[i]]];
}
vector<ll> ids(0);
ll now = 0;
for (auto it = mp.begin(); it != mp.end(); it++) {
ids.push_back(it->first);
it->second = now;
now++;
}
ll len = ids.size();
ll INF = 10000000000000023;
vector<vector<ll>> dists(len, vector<ll>(N, INF));
for (ll i = 0; i < ids.size(); i++) {
dijkstra(graph2, ids[i], dists[i]);
}
// とりあえずこの完全グラフを作る
vector<vector<ll>> dist2(len, vector<ll>(len, INF));
for (ll i = 0; i < len; i++) {
for (ll j = 0; j < len; j++) {
dist2[i][j] = dists[i][ids[j]];
}
}
vector<vector<pll>> num_r(len, vector<pll>(len, make_pair(-1, -1)));
for (ll i = 0; i < K; i++) {
num_r[mp[A[R[i]]]][mp[B[R[i]]]] = make_pair(i, C[R[i]]);
num_r[mp[B[R[i]]]][mp[A[R[i]]]] = make_pair(i, C[R[i]]);
}
// さて、ここからスタート
vector<vector<ll>> ans_dist(len, vector<ll>(1 << K, INF));
ans_dist[0][0] = 0;
priority_queue<pll, vector<pll>, greater<pll>> pque;
pque.push(make_pair(0, 0));
while (pque.size()) {
pll pp = pque.top();
pque.pop();
ll d = pp.first;
ll px = pp.second / (1 << K);
ll py = pp.second % (1 << K);
// debug_out(d, px, py);
if (ans_dist[px][py] < d) {
continue;
}
for (ll i = 0; i < len; i++) {
ll nd = d + dist2[px][i];
ll ny = py;
if (ans_dist[i][ny] > nd) {
ans_dist[i][ny] = nd;
pque.push(make_pair(nd, i * (1 << K) + ny));
}
if (num_r[px][i].first != -1) {
ny |= (1 << num_r[px][i].first);
nd = d + num_r[px][i].second;
if (ans_dist[i][ny] > nd) {
ans_dist[i][ny] = nd;
pque.push(make_pair(nd, i * (1 << K) + ny));
}
}
}
}
cout << ans_dist[len - 1][(1 << K) - 1] << endl;
}