题目传送门 | |
---|---|
树上俩点之间的距离
1号点bfs出每个点的深度d【】数组,离1号点的距离dis数组
x, y 俩点的距离 等于 dis【x】+ dis【y】- 2 * dis【lca(x,y)】;
在bfs过程中 倍增数组
#include<iostream>
#include<queue>
#include<math.h>
using namespace std;
const int N=1e6;
int h[N],ne[N],e[N],w[N],idx;
void add(int a,int b,int c)
{
e[idx]=b;
w[idx]=c;
ne[idx]=h[a];
h[a]=idx++;
}
int n,m,T,t;
int dis[N];//距离数组;
int d[N];//深度数组
int f[N][20];//倍增步数;
void bfs()
{
// 求出dis数组;
queue<int>q;
dis[1]=0;
d[1]=1;
q.push(1);
while(q.size())
{
int x=q.front();
q.pop();
for(int i=h[x];i!=-1;i=ne[i])
{
int y=e[i];
if(d[y]) continue;
d[y] = d[x]+1;
dis[y]=dis[x]+w[i];
//动态规划
f[y][0]=x; //y走一步为x;
for(int k=1;k<=t;k++)
{
f[y][k]=f[f[y][k-1]][k-1];
}
q.push(y);
}
}
}
//返回最近公共祖先;
int lca(int x,int y)
{
if(d[y]>d[x]) swap(x,y);
for(int i=t;i>=0;i--)
{
if(d[f[x][i]]>=d[y]) x=f[x][i];
}
if(x==y) return x;
for(int i=t;i>=0;i--)
{
if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
}
return f[x][0];
}
int main()
{
ios::sync_with_stdio(false);
cin>>T;
while(T--)
{
cin>>n>>m;
t=(int)(log(n)/log(2))+1;
for(int i=1;i<=n;i++)
{
h[i]=-1;
d[i]=0;
}
idx=0;
for(int i=1;i<n;i++)
{
int x,y,z;
cin>>x>>y>>z;
add(x,y,z);
add(y,x,z);
}
bfs();
for(int i=1;i<=m;i++)
{
int x,y;
cin>>x>>y;
cout<<dis[x]+dis[y]-2*dis[lca(x,y)]<<endl;
}
}
return 0;
}