結果

問題 No.898 tri-βutree
ユーザー wakannyaaiwakannyaai
提出日時 2020-02-16 19:34:28
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 596 ms / 4,000 ms
コード長 8,205 bytes
コンパイル時間 2,354 ms
コンパイル使用メモリ 191,100 KB
実行使用メモリ 64,844 KB
最終ジャッジ日時 2024-11-08 23:25:25
合計ジャッジ時間 13,703 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 377 ms
64,844 KB
testcase_01 AC 25 ms
34,560 KB
testcase_02 AC 25 ms
34,560 KB
testcase_03 AC 25 ms
34,688 KB
testcase_04 AC 25 ms
34,560 KB
testcase_05 AC 26 ms
34,688 KB
testcase_06 AC 25 ms
34,560 KB
testcase_07 AC 577 ms
55,452 KB
testcase_08 AC 567 ms
55,576 KB
testcase_09 AC 572 ms
55,580 KB
testcase_10 AC 579 ms
55,576 KB
testcase_11 AC 578 ms
55,448 KB
testcase_12 AC 569 ms
55,444 KB
testcase_13 AC 574 ms
55,448 KB
testcase_14 AC 581 ms
55,448 KB
testcase_15 AC 560 ms
55,572 KB
testcase_16 AC 537 ms
55,580 KB
testcase_17 AC 509 ms
55,444 KB
testcase_18 AC 573 ms
55,448 KB
testcase_19 AC 577 ms
55,456 KB
testcase_20 AC 596 ms
55,576 KB
testcase_21 AC 582 ms
55,580 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
#include <unordered_set>
#define rep(i,n) for(int i = 0; i < n; i++)
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
#define vll vector<vector<long long>>
#define vl vector<long long>
#define vi vector<int>
#define vii vector<vector<int>>
#define pb push_back
#define pf push_front
#define ld long double
#define Sort(a) sort(a.begin(),a.end())
#define cSort(a,cmp) sort(a.begin(),a.end(),cmp)
#define reSort(a) sort(a.rbegin(), a.rend())
static const ll llMAX = numeric_limits<long long>::max();
static const int intMAX = numeric_limits<int>::max();
static const ll  llMIN = numeric_limits<long long>::min();
static const int intMIN = numeric_limits<int>::min();
static const ll d_5 = 100000;
static const ll d9_7 = 1000000007;
static const ll d_9 = 1000000000;
static const double PI=3.14159265358979323846;
template<class T>
void Printvector(std::vector<T> a){
  int size = a.size();
  rep(i,size){
    cout<<a[i]<<" ";
  }
  cout<<endl;
}
template<class T>
void Printvector(std::vector<std::vector<T>> a){
  int size = a.size();
  rep(i,size){
    int size2=a[i].size();
    rep(j,size2){
      cout<<a[i][j]<<" ";
    }
    cout<<endl;
  }
  cout<<endl;
}
ll digitpower(ll a,ll b){//aのb乗を計算
  if(b==1){
    return a;
  }else if(b==0){
    return 1;
  }
  int mode=0;
  if(b%2==1){
    ll tmp = digitpower(a,(b-1)/2);
    if(mode==1){
      tmp%=d9_7;
    }
    tmp*=tmp;
    if(mode==1){
      tmp%=d9_7;
    }
    tmp*=a;
    if(mode==1){
      return tmp%d9_7;
    }else{
      return tmp;
    }
  }else{
    ll tmp = digitpower(a,(b)/2);
    if(mode==1){
      tmp%=d9_7;
    }
    tmp*=tmp;
    if(mode==1){
      tmp%=d9_7;
    }
    if(mode==1){
      return tmp%d9_7;
    }else{
      return tmp;
    }
  }
}
vl facs(2000010,-1);
ll Factrial(ll num){
  if(facs[num]!=-1){
    return facs[num];
  }
  if(num==1||num<=0){
    return 1;
  }else if(num<0){
    printf("ERROR_minus\n");
    return 0;
  }else{
    facs[num]=(num*Factrial(num-1))%d9_7;
    return facs[num];
  }
}
long long modinv(long long a, long long m) {//modの逆元

    long long b = m, u = 1, v = 0;
    while (b) {
        long long t = a / b;
        a -= t * b; swap(a, b);
        u -= t * v; swap(u, v);
    }
    u %= m; 
    if (u < 0) u += m;
    return u;
}

vl invs(2000010,-1);
ll linercomb(ll n,ll k, ll mod){//n,kの線形時間で求める
if(n<k)return 0;
if(n<0)return 0;
if(k==0 || k==n)return 1;
  ll ans=Factrial(n);
  if(invs[k]==-1){
    invs[k]=modinv(Factrial(k),mod);
  }
  ans*=invs[k];
  ans%=d9_7;
  ll k1=Factrial(n-k);
  k1%=mod;
  ans*=modinv(k1,mod);
  ans%=mod;
  return ans;
}
unordered_map<ll,ll> prime_factor(int64_t n) {
  unordered_map<ll,ll> ret;
  for(int64_t i = 2; i * i <= n; i++) {
    while(n % i == 0) {
      ret[i]++;
      n /= i;
    }
  }
  if(n != 1) ret[n] = 1;
  return ret;
}

template<class T>
vector<T> getaccum(vector<T> a){
  int size=a.size();
  vector<T> ans(size);
  ans[0]=a[0];
  for(int i=0;i<size-1;i++){
    ans[i+1]=ans[i]+a[i+1];
    //ans[i+1]%=d9_7;
  }
  return ans;
}

ll getaccumnum(vector<ll> accum,int l,int r){//閉区間[l,r]の総和
  if(l==0){
    return accum[r];
  }else{
    return accum[r]-accum[l-1];
  }
}
//Segment Tree
template<class T>
class SegmentTree{
  public:
   int n;
   vector<T> nodes;
    //constructor
    SegmentTree(){
      ;
    }
    SegmentTree(int size,T init){
      initialize(size,init);
    }
    //関数を定義(minなのかmaxなのかとか)
    T thisoperator(T a, T b){
      return min(a,b);
    }
    void update(int x,T a){//xがindex
      x+=n-1;
      nodes[x]=a;
      while(x>0){
        x=(x-1)/2;
        nodes[x]=thisoperator(nodes[2*x+1],nodes[2*x+2]);
      }
    }
   void initialize(int inputn,T init){
      int k=0;
      while(inputn>(1<<k)){
        k++;
      }
      n=(1<<k);
      nodes=vector<T> ((1<<k)*2-1,init);
   }
   ll sec_get(int reql,int reqr,int nowindex=0,int nowl=0,int nowr=-1){
     // 最初に呼び出されたときの対象区間は [0, n)
    if(nowr < 0) nowr = n;

    // 要求区間と対象区間が交わらない -> 適当に返す 
    //default に注意!
    if(nowr <= reql || reqr <= nowl) return (T)1<<60;

    // 要求区間が対象区間を完全に被覆 -> 対象区間を答えの計算に使う
    if(reql <= nowl && nowr <= reqr) return nodes[nowindex];
    // 要求区間が対象区間の一部を被覆 -> 子について探索を行う
    // 左側の子を vl ・ 右側の子を vr としている
    // 新しい対象区間は、現在の対象区間を半分に割ったもの
    T val1 = sec_get(reql, reqr, 2*nowindex+1, nowl, (nowl+nowr)/2);
    T val2 = sec_get(reql, reqr, 2*nowindex+2, (nowl+nowr)/2, nowr);
    return thisoperator(val1, val2);
   }
   void Printn(){
     cout<<"Printvector Seg"<<endl;
     rep(i,n){
       cout<<nodes[i+n-1]<<" ";
     }
     cout<<endl;
   }
};
template<class T>
class LCA{
  public:
   int n;//総ノード数
   struct LCA_data{
     int index;
     T depth;
   };
   SegmentTree<T> seg;
   //DFSでグラフTを根から探索した際の頂点の訪問順を並べた列S=(v0,v1,...,vM−1)
   vector<LCA_data> nodes;
   vector<int> Firstloc;
    //constructor
    LCA(int size){
      initialize(size);
    }
    
   void initialize(int inputn){
     n = inputn;
     Firstloc=vector<int> (n,300000);
   }
   void Printn(){
     cout<<"Printvector lca index"<<endl;
     rep(i,nodes.size()){
       cout<<nodes[i].index<<" ";
     }
     cout<<endl;
   }
   void Printd(){
     cout<<"Printvector lca depth"<<endl;
     rep(i,nodes.size()){
       cout<<nodes[i].depth<<" ";
     }
     cout<<endl;
   }
   void visitnode(int index,T depth){//オイラーツアー実行用
     nodes.push_back({index,depth});
     Firstloc[index]=min(Firstloc[index],(int)nodes.size()-1);
   }
   T getLCAdepth(int a,int b){//深さを返す、閉区間指定
     int l=Firstloc[a];
     int r=Firstloc[b];
     if(l>r)swap(l,r);
     return seg.sec_get(l,r);
   }
   void EndpreDFS(){
     //セグ木を作る
     n=nodes.size();
     seg.initialize(n,(T)1<<60);
     int cnt=0;
     for(auto i:nodes){
       seg.update(cnt,nodes[cnt].depth);
       cnt++;
     }
   }
   T getdepth(int index){
     return nodes[Firstloc[index]].depth;
   }
};
//グラフ
LCA<ll> lca(0);
struct edge{
  int to;
  ll length;
  int num;
  int pair;
};
struct treev{
  int num;
  vector<int> nexts;
};
vector<treev> vs;
vector<edge> es;
 
int cmax=0;
void createtreenode(int num){//引数は頂点の数
  vs=vector<treev>(num);
  /*treev* newt;
  try{
      treev* newt=new treev;
  }
  catch (std::bad_alloc&) {
      // メモリ確保に失敗
      // エラー処理
  }*/
  rep(i,num){
    vs[i].num=i;
  }
  return ;
}
void insertree_bidire(ll len,int Va,int Vb,int num){//結ぶ頂点を
  //双方向
  es.pb({Va,len,num,-1});
  es.pb({Vb,len,num,-1});
  int size=es.size();
  es[size-1].pair=size-2;
  es[size-2].pair=size-1;
  vs[Va].nexts.push_back(size-1);
  vs[Vb].nexts.push_back(size-2);
  
  return;
}

void DFS(int root,ll depth,int pre){
  //ここにやりたい処理を書く
  //cout<<root<<endl;
  lca.visitnode(root,depth);
  for(auto i:vs[root].nexts){
    if(pre==es[i].to){
      continue;
    }
    
    DFS(es[i].to,depth+es[i].length,root);
    lca.visitnode(root,depth);
  }
  return;
}

struct datas{
  int a;
  int b;
};/*
bool cmp(const datas &a, const datas &b)
{
    return a.num < b.num;
}*/
//Segment Tree

int main(void){
  int n;
  cin>>n;
  vl u(n-1),v(n-1),w(n-1);
  rep(i,n-1){
    cin>>u[i]>>v[i]>>w[i];
  }
  createtreenode(n);
  rep(i,n-1){
    insertree_bidire(w[i],u[i],v[i],i);
  }
  lca.initialize(n);
  DFS(0,0,-1);
  lca.EndpreDFS();
  int q;
  cin>>q;
  rep(i,q){
    int x,y,z;
    cin>>x>>y>>z;
    ll n1=lca.getdepth(x)+lca.getdepth(y)-2*lca.getLCAdepth(x,y);
    ll n2=lca.getdepth(x)+lca.getdepth(z)-2*lca.getLCAdepth(x,z);
    ll n3=lca.getdepth(z)+lca.getdepth(y)-2*lca.getLCAdepth(y,z);
    cout<<(n1+n2+n3)/2<<endl;
  }
  return 0;
}

//<<std::setprecision(30)

//重複削除
 /* std::sort(vec.begin(), vec.end());
  vec.erase(std::unique(vec.begin(), vec.end()), vec.end());*/
0