제출 #554561

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
554561		Soumya1	Cities (BOI16_cities)	C++17	100 / 100	2681 ms	44880 KiB

cities

#include <bits/stdc++.h>
#ifdef __LOCAL__
#include <debug_local.h>
#endif
using namespace std;
const int mxN = 1e5 + 5;
const long long inf = 1e18;
using ll = long long;
int idx[mxN];
vector<pair<int, int>> ad[mxN];
vector<int> terminals;
ll d[mxN][32];
void testCase() {
  int n, k, m;
  cin >> n >> k >> m;
  for (int i = 0; i < k; i++) {
    int x;
    cin >> x;
    terminals.push_back(x);
    idx[x] = i;
  }
  while (m--) {
    int u, v, w;
    cin >> u >> v >> w;
    ad[u].push_back({v, w});
    ad[v].push_back({u, w});
  }
  for (int i = 1; i <= n; i++) {
    for (int mask = 0; mask < (1 << k); mask++) d[i][mask] = inf;
  }
  for (int t : terminals) d[t][1 << idx[t]] = 0;
  for (int mask = 1; mask < (1 << k); mask++) {
    for (int i = 1; i <= n; i++) {
      for (int s_mask = mask & (mask - 1); s_mask > 0; s_mask = mask & (s_mask - 1)) {
        d[i][mask] = min(d[i][mask], d[i][s_mask] + d[i][mask ^ s_mask]);
      }
    }
    priority_queue<pair<ll, int>> pq;
    for (int i = 1; i <= n; i++) pq.push({-d[i][mask], i});
    while (!pq.empty()) {
      auto [_, u] = pq.top();
      pq.pop();
      if (d[u][mask] != -_) continue;
      for (auto [v, w] : ad[u]) {
        if (d[v][mask] > d[u][mask] + w) {
          d[v][mask] = d[u][mask] + w;
          pq.push({-d[v][mask], v});
        }
      }
    }
  }
  ll ans = inf;
  for (int i = 1; i <= n; i++) ans = min(ans, d[i][(1 << k) - 1]);
  cout << ans << "\n";
}
int main() {
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  testCase();
  return 0;
}

• Subtask 1: 22 / 22
• Subtask 2: 14 / 14
• Subtask 3: 15 / 15
• Subtask 4: 23 / 23
• Subtask 5: 26 / 26