题目描述 Description
小机房有棵焕狗种的树,树上有N个节点,节点标号为0到N-1,有两只虫子名叫飘狗和大吉狗,分居在两个不同的节点上。有一天,他们想爬到一个节点上去搞基,但是作为两只虫子,他们不想花费太多精力。已知从某个节点爬到其父亲节点要花费 c 的能量(从父亲节点爬到此节点也相同),他们想找出一条花费精力最短的路,以使得搞基的时候精力旺盛,他们找到你要你设计一个程序来找到这条路,要求你告诉他们最少需要花费多少精力
输入描述 Input Description
第一行一个n,接下来n-1行每一行有三个整数u,v, c 。表示节点 u 爬到节点 v 需要花费 c 的精力。
第n+1行有一个整数m表示有m次询问。接下来m行每一行有两个整数 u ,v 表示两只虫子所在的节点
输出描述 Output Description
一共有m行,每一行一个整数,表示对于该次询问所得出的最短距离。
样例输入 Sample Input
3
1 0 1
2 0 1
3
1 0
2 0
1 2
样例输出 Sample Output
1
1
2
数据范围及提示 Data Size & Hint
1<=n<=50000, 1<=m<=75000, 0<=c<=1000
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxx = 50000 + 100;
int head[maxx],deep[maxx];//shendu
int anc[maxx][20+2],dis[maxx][20+2];//zuxian yugenjiediandejvli
bool done[maxx];
int n,m,x,y,z,num,root;
struct Edge{
int next;//qidian
int to;//zhongdian
int value;//jingli
}Edges[maxx<<1];
void build(int x,int y,int z){ //建边
Edges[++num].to=y;
Edges[num].next=head[x];
//cout<<head[x]<<" ";
Edges[num].value=z;
head[x]=num;
//cout<<num<<" ";
}
void Dfs(int x){
done[x]=true;
for(int i=1;i<=20;i++){
if((1<<i) > deep[x]) break;
anc[x][i]=anc[anc[x][i-1]][i-1];
dis[x][i]=dis[x][i-1]+dis[anc[x][i-1]][i-1];
}
for(int i=head[x];i;i=Edges[i].next){
int now=Edges[i].to;
if(done[now]) continue;
deep[now]=deep[x]+1;
anc[now][0]=x;
dis[now][0]=Edges[i].value;
Dfs(now);
}
}
inline int Lca(int x,int y){
int Ans=0;
if(deep[x]>deep[y]) swap(x,y);
int d=deep[y]-deep[x];//d:深度差
for(int i=0;i<=20;i++){
if((1<<i) & d) { //........................
Ans += dis[y][i];
y=anc[y][i];
}
}
for(int i=20;i>=0;i--){
if(anc[x][i]!=anc[y][i]){
Ans += dis[x][i];
Ans += dis[y][i];
x=anc[x][i];
y=anc[y][i];
}
}
if(x==y) return Ans;
else{
Ans+=dis[x][0]+dis[y][0];
return Ans;
}
}
int main(){
cin>>n;
for(int i=1;i<n;i++){
cin>>x>>y>>z;//qidian zhongdian jingli
build(x,y,z); build(y,x,z);//建双向边
}
Dfs(0);//初始化 ,以0为根节点
cin>>m;
while( m-- ){//多组询问
cin>>x>>y;
printf("%d\n",Lca(x,y));
}
return 0;
}