結果

問題 No.898 tri-βutree
ユーザー saxofone111saxofone111
提出日時 2021-03-11 23:34:42
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 438 ms / 4,000 ms
コード長 2,527 bytes
コンパイル時間 2,619 ms
コンパイル使用メモリ 210,192 KB
実行使用メモリ 24,668 KB
最終ジャッジ日時 2024-11-08 23:52:45
合計ジャッジ時間 11,579 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 300 ms
24,668 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 3 ms
5,248 KB
testcase_03 AC 3 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 427 ms
17,300 KB
testcase_08 AC 438 ms
17,300 KB
testcase_09 AC 436 ms
17,296 KB
testcase_10 AC 438 ms
17,424 KB
testcase_11 AC 431 ms
17,424 KB
testcase_12 AC 430 ms
17,416 KB
testcase_13 AC 429 ms
17,296 KB
testcase_14 AC 433 ms
17,300 KB
testcase_15 AC 425 ms
17,304 KB
testcase_16 AC 425 ms
17,292 KB
testcase_17 AC 425 ms
17,424 KB
testcase_18 AC 428 ms
17,296 KB
testcase_19 AC 424 ms
17,424 KB
testcase_20 AC 429 ms
17,296 KB
testcase_21 AC 430 ms
17,292 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"

#define MOD 1000000007
#define rep(i, n) for(ll i=0; i < (n); i++)
#define rrep(i, n) for(ll i=(n)-1; i >=0; i--)
#define ALL(v) v.begin(),v.end()
#define rALL(v) v.rbegin(),v.rend()
#define FOR(i, j, k) for(ll i=j;i<k;i++)
#define debug_print(var) cerr << #var << "=" << var <<endl;
#define DUMP(i, v)for(ll i=0;i<v.size();i++)cerr<<v[i]<<" "
#define fi first
#define se second

using namespace std;
typedef long long int ll;
typedef vector<ll> llvec;
typedef vector<double> dvec;
typedef pair<ll, ll> P;
typedef long double ld;
struct edge{ll x, c;};

vector<vector<edge>> e;

struct segment_tree{
  ll N;
  llvec v;
  ll init=5e18;//initial value
  ll f(ll a, ll b){ //function
    return min(a, b);
  }
  segment_tree(ll n){
    N=1;
    while(N<n){
      N*=2;
    }
    v = llvec(2*N-1, init);
  }
  
  void set(ll i, ll val){
    i += N-1;
    v[i] = val;
    while(i>0){
      i = (i-1)/2;
      v[i] = f(v[i*2+1], v[i*2+2]);
    }
  }
  void add(ll i, ll val){
    i += N-1;
    v[i] += val;
    while(i>0){
      i = (i-1)/2;
      v[i] = f(v[i*2+1], v[i*2+2]);
    }
  }

  ll get(ll L, ll R){// L <= i < R
    L += N-1;
    R += N-1;
    ll vl = init;
    ll vr = init;
    while(L<R){
      if(L%2==0){
        vl = f(vl, v[L]);
        L++;
      }
      if(R%2==0){
        vr = f(vr, v[R-1]);
        R--;
      }
      R=(R-1)/2;
      L=(L-1)/2;
    }
    return f(vl, vr);
  }

  ll operator[](ll i){
    return v[i+N-1];
  }
  
};

llvec a;
llvec v;

void dfs(ll from, ll to, ll c){
  v[to] = a.size();
  a.push_back(c);
  for(auto ie: e[to]){
    if(ie.x == from)continue;
    dfs(to, ie.x, c+ie.c);
    a.push_back(c);
  }
  return;
}

/**************************************
** A main function starts from here  **
***************************************/
int main(){
  ll N;
  cin >> N;
  e = vector<vector<edge>>(N);
  rep(i, N-1){
    ll u, v, c;
    cin >> u >> v >> c;
    //u--;v--;
    e[u].push_back({v, c});
    e[v].push_back({u, c});
  }
  v = llvec(N);
  dfs(-1, 0, 0);
  segment_tree sg(a.size());
  rep(i, a.size()){
    sg.set(i, a[i]);
  }

  ll Q;
  cin >> Q;
  while(Q--){
    ll x, y, z;
    cin >> x >> y >> z;
    if(v[x]>v[y]){
      swap(x, y);
    }
    if(v[y]>v[z]){
      swap(y, z);
    }
    if(v[x]>v[y]){
      swap(x, y);
    }
    ll a = sg.get(v[x], v[y]+1);
    ll b = sg.get(v[y], v[z]+1);
    ll A = sg[v[x]] + sg[v[y]] - 2*a;
    ll B = sg[v[z]] + sg[v[y]] - 2*b;
    cout << A + B - ( sg[v[y]] - max(a, b)) << endl;  
  }
  
  return 0;
}
0