結果

問題 No.1442 I-wate Shortest Path Problem
ユーザー SHIJOUSHIJOU
提出日時 2021-04-02 22:43:54
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 580 ms / 3,000 ms
コード長 3,579 bytes
コンパイル時間 2,793 ms
コンパイル使用メモリ 228,088 KB
実行使用メモリ 48,408 KB
最終ジャッジ日時 2024-06-06 07:31:47
合計ジャッジ時間 11,593 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,940 KB
testcase_02 AC 8 ms
6,940 KB
testcase_03 AC 36 ms
6,944 KB
testcase_04 AC 7 ms
6,940 KB
testcase_05 AC 6 ms
6,940 KB
testcase_06 AC 36 ms
6,940 KB
testcase_07 AC 5 ms
6,940 KB
testcase_08 AC 32 ms
6,940 KB
testcase_09 AC 13 ms
6,944 KB
testcase_10 AC 38 ms
6,940 KB
testcase_11 AC 38 ms
6,940 KB
testcase_12 AC 394 ms
42,108 KB
testcase_13 AC 217 ms
33,536 KB
testcase_14 AC 294 ms
38,716 KB
testcase_15 AC 276 ms
35,968 KB
testcase_16 AC 351 ms
38,420 KB
testcase_17 AC 543 ms
43,032 KB
testcase_18 AC 523 ms
43,208 KB
testcase_19 AC 459 ms
41,876 KB
testcase_20 AC 577 ms
43,176 KB
testcase_21 AC 580 ms
43,280 KB
testcase_22 AC 224 ms
39,168 KB
testcase_23 AC 428 ms
48,408 KB
testcase_24 AC 176 ms
34,344 KB
testcase_25 AC 350 ms
43,940 KB
testcase_26 AC 170 ms
39,024 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

//#define _GLIBCXX_DEBUG
#include <bits/stdc++.h>
using namespace std;
#define rep(i, n) for(int i=0; i<n; ++i)
#define all(v) v.begin(), v.end()
#define rall(v) v.rbegin(), v.rend()
using ll = int64_t;
using ull = uint64_t;
using ld = long double;
using P = pair<int, int>;
using vs = vector<string>;
using vi = vector<int>;
using vvi = vector<vi>;
template<class T> using PQ = priority_queue<T>;
template<class T> using PQG = priority_queue<T, vector<T>, greater<T>>;
const int INF = 0xccccccc;
const ll LINF = 0xcccccccccccccccLL;
template<typename T1, typename T2>
inline bool chmax(T1 &a, T2 b) {return a < b && (a = b, true);}
template<typename T1, typename T2>
inline bool chmin(T1 &a, T2 b) {return a > b && (a = b, true);}
template<typename T1, typename T2>
istream &operator>>(istream &is, pair<T1, T2> &p) { return is >> p.first >> p.second;}
template<typename T1, typename T2>
ostream &operator<<(ostream &os, const pair<T1, T2> &p) { return os << p.first << ' ' << p.second;}

template<class T>
struct lca {
  int n, root, l;
  vector<vector<int> > to, par;
  vector<vector<T> > co;
  vector<int> dep;
  vector<T> costs;
  lca(int n):n(n), to(n), co(n), dep(n), costs(n) {
    l = 0;
    while((1<<l) < n) ++l;
    par = vector<vector<int> >(n+1, vector<int>(l, n));
  }
  void add(int a, int b, T c=0) {
    to[a].push_back(b); co[a].push_back(c);
    to[b].push_back(a); co[b].push_back(c);
  }
  void dfs(int v, int d=0, T c=0, int p=-1) {
    if(p != -1) par[v][0] = p;
    dep[v] = d;
    costs[v] = c;
    for(int i = 0; i < to[v].size(); ++i) {
      int u = to[v][i];
      if(u == p) continue;
      dfs(u, d+1, c+co[v][i], v);
    }
  }
  void init(int _root=0) {
    root = _root;
    dfs(root);
    for(int i = 0; i < l-1; ++i) {
      for(int v = 0; v < n; ++v) {
        par[v][i+1] = par[par[v][i]][i];
      }
    }
  }
  int lca_(int a, int b) {
    if(dep[a] > dep[b]) swap(a, b);
    int gap = dep[b]-dep[a];
    for(int i = l-1; i >= 0; --i) {
      int len = 1<<i;
      if(gap >= len) {
        gap -= len;
        b = par[b][i];
      }
    }
    if(a == b) return a;
    for(int i = l-1; i >= 0; --i) {
      int na = par[a][i];
      int nb = par[b][i];
      if(na != nb) {
        a = na;
        b = nb;
      }
    }
    return par[a][0];
  }
  int length(int a, int b) {
    int c = lca_(a, b);
    return dep[a]+dep[b]-dep[c]*2;
  }
  T dist(int a, int b) {
    int c = lca_(a, b);
    return costs[a]+costs[b]-costs[c]*2;
  }
};

//head



int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  int n, k;
  cin >> n >> k;
  vector<vector<pair<int, ll>>> G(n+k);
  lca<ll> l(n);
  rep(i, n-1) {
    int a, b, c;
    cin >> a >> b >> c;
    a--; b--;
    G[a].emplace_back(b, c);
    G[b].emplace_back(a, c);
    l.add(a, b, c);
  }
  l.init();
  vi _(k);
  rep(i, k) {
    int m, p;
    cin >> m >> p;
    _[i] = p;
    rep(j, m) {
      int x;
      cin >> x;
      x--;
      G[x].emplace_back(n+i, p);
      G[n+i].emplace_back(x, 0);
    }
  }
  PQG<pair<ll, int>> que;
  vector<vector<ll>> dist(k, vector<ll>(n+k, LINF));
  rep(i, k) {
    que.emplace(0, n+i);
    dist[i][n+i] = 0;
    while(not que.empty()) {
      auto [d, now] = que.top();
      que.pop();
      if(dist[i][now] != d) continue;
      for(auto [to, c]:G[now]) {
        if(chmin(dist[i][to], d+c)) {
          que.emplace(d+c, to);
        }
      }
    }
  }
  int q;
  cin >> q;
  while(q--) {
    int u, v;
    cin >> u >> v;
    u--; v--;
    ll ans = l.dist(u, v);
    rep(i, k) chmin(ans, dist[i][u]+dist[i][v]+_[i]);
    cout << ans << '\n';
  }
}
0