結果
| 問題 | No.898 tri-βutree |
| ユーザー |
 wakannyaai
|
| 提出日時 | 2020-02-16 17:32:40 |
| 言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
| 結果 |
WA
|
| 実行時間 | - |
| コード長 | 8,206 bytes |
| コンパイル時間 | 2,114 ms |
| コンパイル使用メモリ | 190,976 KB |
| 実行使用メモリ | 64,848 KB |
| 最終ジャッジ日時 | 2024-10-06 14:34:53 |
| 合計ジャッジ時間 | 11,873 ms |
|
ジャッジサーバーID （参考情報） |
judge4 / judge5 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| 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 | - |
ソースコード
#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);
}
T 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 llMAX/100;
// 要求区間が対象区間を完全に被覆 -> 対象区間を答えの計算に使う
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,n+100);
}
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,llMAX/100);
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());*/