結果

問題 No.898 tri-βutree
ユーザー wakannyaaiwakannyaai
提出日時 2020-02-16 19:33:18
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 5,838 bytes
コンパイル時間 1,948 ms
コンパイル使用メモリ 184,816 KB
実行使用メモリ 33,680 KB
最終ジャッジ日時 2024-10-06 14:39:34
合計ジャッジ時間 9,356 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 WA -
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 WA -
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 WA -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
権限があれば一括ダウンロードができます

ソースコード

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;

//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,nodes.size());
   }
   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