题目描述
给出一棵带有边权的树,问两点之间的距离。
输入格式
第一行两个整数 n 和 m ,分别表示点数和询问数。
接下来 n-1 行,每行三个整数 x,y,z,表示 x 与 y 通过一条权为 z 的边连接。
接下来 m 行,每行两个整数 x,y,代表一组询问。
##输出格式
输出 m 行,每行一个整数,对应一组询问的答案。
样例数据 1
输入
3 3
1 2 1
1 3 2
1 2
1 3
2 3
输出
1
2
3
备注
【数据范围】
对 30% 的输入数据 :1≤n,m≤1000
对 100% 的输入数据 :1≤n,m≤100000;1≤z≤10000
代码
dfs_:处理点的深度,距离以及预处理跳
2
i
2^i
2i步后到达的点
LCA:倍增找公共祖先
注意:LCA_中
f
o
r
for
for循环是从
20
−
0
20-0
20−0不是
20
−
1
20-1
20−1也不是
0
−
20
o
r
1
−
20
0-20or1-20
0−20or1−20
#include <bits/stdc++.h>
using namespace std;
const int N=1e5+50;
int f[N][30],dis[N],tot,dep[N],fir[N];
struct node{
int to,nxt,w;
}e[N<<1];
inline int read(){
int cnt=0,f=1;
char c=getchar();
while(!isdigit(c)){
if(c=='-'){
f=-f;
}
c=getchar();
}
while(isdigit(c)){
cnt=(cnt<<3)+(cnt<<1)+(c^48);
c=getchar();
}
return cnt*f;
}
void add(int x,int y,int z){
e[++tot].nxt=fir[x];fir[x]=tot;e[tot].to=y;e[tot].w=z;
}
void dfs_(int u,int fa){
dep[u]=dep[fa]+1;
f[u][0]=fa;
for(int i=1;i<=20;i++){
f[u][i]=f[f[u][i-1]][i-1];
}
for(int i=fir[u];i;i=e[i].nxt){
int v=e[i].to;
if(v==fa) continue;
dis[v]=dis[u]+e[i].w;
dfs_(v,u);
}
}
int LCA_(int u,int v){
if(dep[u]<dep[v]) swap(u,v);
for(int i=20;i>=0;i--){
if(dep[f[u][i]]>=dep[v]) u=f[u][i];
}
if(u==v) return v;
for(int i=20;i>=0;i--){
if(f[u][i]!=f[v][i]){
u=f[u][i];v=f[v][i];
}
}
return f[u][0];
}
void solve(){
int n,m;
n=read(),m=read();
for(int i=1;i<n;i++){
int x,y,z;x=read(),y=read(),z=read();
add(x,y,z);add(y,x,z);
}
dfs_(1,0);
for(int i=1;i<=m;i++){
int x,y;x=read(),y=read();
int mid=LCA_(x,y);
int disn=dis[x]+dis[y]-2*dis[mid];
printf("%d\n",disn);
}
}
int main(){
solve();
return 0;
}