結果

問題 No.1812 Uribo Road
ユーザー KudeKude
提出日時 2022-01-14 22:21:23
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
MLE  
実行時間 -
コード長 1,979 bytes
コンパイル時間 8,682 ms
コンパイル使用メモリ 293,808 KB
最終ジャッジ日時 2025-01-27 11:45:33
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 21 TLE * 8 MLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
namespace {
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-function"
#include<atcoder/all>
#pragma GCC diagnostic pop
using namespace std;
using namespace atcoder;
#define rep(i,n)for (int i = 0; i < int(n); ++i)
#define rrep(i,n)for (int i = int(n)-1; i >= 0; --i)
#define all(x) (x).begin(), (x).end()
#define rall(x) (x).rbegin(), (x).rend()
template<class T> void chmax(T& a, const T& b) { a = max(a, b); }
template<class T> void chmin(T& a, const T& b) { a = min(a, b); }
using ll = long long;
using P = pair<int,int>;
using VI = vector<int>;
using VVI = vector<VI>;
using VL = vector<ll>;
using VVL = vector<VL>;

} int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  int n, m, k;
  cin >> n >> m >> k;
  // [state][v]
  struct Edge {
    int to;
    int c;
    int k;
  };
  vector<vector<Edge>> to(n);
  VI r(k);
  rep(i, k) cin >> r[i], r[i]--;
  sort(all(r));
  int nxt = 0;
  rep(i, m) {
    int a, b, c;
    cin >> a >> b >> c;
    a--, b--;
    if (nxt < k && r[nxt] == i) {
      to[a].emplace_back(Edge{b, c, nxt});
      to[b].emplace_back(Edge{a, c, nxt});
      nxt++;
    } else {
      to[a].emplace_back(Edge{b, c, -1});
      to[b].emplace_back(Edge{a, c, -1});
    }
  }
  assert(nxt == k);
  constexpr ll INF = 1002003004005006007;
  vector<ll> dist(n << k, INF);
  auto cmp = [](const pair<int, ll>& x, const pair<int, ll>& y) {
    return x.second > y.second;
  };
  priority_queue<pair<int, ll>, vector<pair<int, ll>>, decltype(cmp)> q(cmp);
  auto push = [&](int s, int v, ll d) {
    int idx = s | v << k;
    if (d < dist[idx]) {
      dist[idx] = d;
      q.emplace(idx, d);
    }
  };
  push(0, 0, 0);
  while(!q.empty()) {
    auto [idx, d] = q.top(); q.pop();
    if (dist[idx] != d) continue;
    int u = idx >> k;
    int s = idx & (1 << k) - 1;
    for(auto [v, c, b] : to[u]) {
      int ns = b != -1 ? s | 1 << b : s;
      push(ns, v, d + c);
    }
  }
  cout << dist.back() << '\n';
}
0