結果

問題 No.1094 木登り / Climbing tree
ユーザー tnakao0123tnakao0123
提出日時 2020-06-24 19:02:29
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 338 ms / 2,000 ms
コード長 2,058 bytes
コンパイル時間 1,147 ms
コンパイル使用メモリ 103,584 KB
実行使用メモリ 44,708 KB
最終ジャッジ日時 2024-04-25 18:25:08
合計ジャッジ時間 11,351 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
12,136 KB
testcase_01 AC 285 ms
44,592 KB
testcase_02 AC 100 ms
43,624 KB
testcase_03 AC 51 ms
15,248 KB
testcase_04 AC 80 ms
30,384 KB
testcase_05 AC 144 ms
43,572 KB
testcase_06 AC 127 ms
23,040 KB
testcase_07 AC 280 ms
44,684 KB
testcase_08 AC 306 ms
44,576 KB
testcase_09 AC 302 ms
44,480 KB
testcase_10 AC 311 ms
44,556 KB
testcase_11 AC 299 ms
44,708 KB
testcase_12 AC 326 ms
44,540 KB
testcase_13 AC 327 ms
44,656 KB
testcase_14 AC 326 ms
44,696 KB
testcase_15 AC 140 ms
20,192 KB
testcase_16 AC 269 ms
37,904 KB
testcase_17 AC 183 ms
27,908 KB
testcase_18 AC 175 ms
25,144 KB
testcase_19 AC 272 ms
35,336 KB
testcase_20 AC 338 ms
44,600 KB
testcase_21 AC 228 ms
28,116 KB
testcase_22 AC 335 ms
44,636 KB
testcase_23 AC 319 ms
44,684 KB
testcase_24 AC 335 ms
44,608 KB
testcase_25 AC 321 ms
44,556 KB
testcase_26 AC 322 ms
44,580 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

/* -*- coding: utf-8 -*-
 *
 * 1094.cc:  No.1094 木登り / Climbing tree - yukicoder
 */

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<list>
#include<queue>
#include<deque>
#include<algorithm>
#include<numeric>
#include<utility>
#include<complex>
#include<functional>
 
using namespace std;

/* constant */

const int MAX_N = 200000;
const int BN = 18;

/* typedef */

typedef queue<int> qi;
typedef pair<int,int> pii;
typedef vector<pii> vpii;

/* global variables */

vpii nbrs[MAX_N];
int ds[MAX_N], ps[MAX_N][BN], cs[MAX_N][BN];

/* subroutines */

int pcost(int u, int v) {
  int c = 0;
  if (ds[u] > ds[v]) swap(u, v);

  for (int k = BN - 1; k >= 0; k--)
    if (((ds[v] - ds[u]) >> k) & 1) {
      c += cs[v][k];
      v = ps[v][k];
    }
  if (u == v) return c;

  for(int k = BN - 1; k >= 0; k--)
    if (ps[u][k] != ps[v][k]) {
      c += cs[u][k] + cs[v][k];
      u = ps[u][k];
      v = ps[v][k];
    }
  
  return c + cs[u][0] + cs[v][0];
}

/* main */

int main() {
  int n;
  scanf("%d", &n);

  for (int i = 1; i < n; i++) {
    int a, b, c;
    scanf("%d%d%d", &a, &b, &c);
    a--, b--;
    nbrs[a].push_back(pii(b, c));
    nbrs[b].push_back(pii(a, c));
  }

  ds[0] = 0, ps[0][0] = -1, cs[0][0] = 0;
  qi q;
  q.push(0);

  while (! q.empty()) {
    int u = q.front(); q.pop();

    for (int k = 0; k < BN - 1; k++) {
      if (ps[u][k] >= 0) {
	ps[u][k + 1] = ps[ps[u][k]][k];
	cs[u][k + 1] = cs[u][k] + cs[ps[u][k]][k];
      }
      else
	ps[u][k + 1] = -1;
    }

    int up = ps[u][0], vd = ds[u] + 1;
    vpii &nbru = nbrs[u];
    for (vpii::iterator vit = nbru.begin(); vit != nbru.end(); vit++) {
      int v = vit->first;
      if (v != up) {
	ds[v] = vd, ps[v][0] = u, cs[v][0] = vit->second;
	q.push(v);
      }
    }
  }

  int qn;
  scanf("%d", &qn);
  while (qn--) {
    int s, t;
    scanf("%d%d", &s, &t);
    s--, t--;

    int c = pcost(s, t);
    printf("%d\n", c);
  }

  return 0;
}
0