a,b两点间的距离为 dis[a]+dis[b]-2*dis[lca(a,b)] ,dis为到根节点的距离
#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
const int maxn=40010;
struct node
{
int to,w;
node(int a,int b) //构造函数
{
to=a;
w=b;
}
};
vector<node>g[maxn];
int fa[maxn][17]; //表示节点i往上跳2^j次后的节点 注意j的取值范围
int d[maxn]; //当前节点所在的深度
int n,m;
int dis[maxn]; //到根节点的距离
bool vis[maxn];
int root; //根结点
void init() //初始化
{
memset(fa,0,sizeof(fa));
memset(vis,0,sizeof(vis));
memset(d,0,sizeof(d));
memset(dis,0,sizeof(dis));
}
void dfs(int u)
{
int i;
vis[u]=1;
for(i=0;i<g[u].size();i++) //每一个与u相连的节点i
{
int v=g[u][i].to;
if(!vis[v])
{
d[v]=d[u]+1;
dis[v]=dis[u]+g[u][i].w; //更新到根节点的距离
dfs(v);
}
}
}//深搜出各点的深度,存在d中
void bz()
{
int i,j;
for(j=1;j<=16;j++)
for(i=1;i<=n;i++)
fa[i][j]=fa[fa[i][j-1]][j-1]; //i的第2^j祖先就是i的第2^(j-1)祖先的第2^(j-1)祖先
//i的父辈们的fa数组都已经计算完毕,所以可以用来计算i
}
int lca(int u,int v)
{
if(d[u]<d[v])
swap(u,v); //保证深度大的点为u,方便操作
int dc=d[u]-d[v]; //计算深度差
int i;
for(i=0;i<=16;i++) //值得注意的是,这里需要从零枚举
{ //此循环用于提到深度相同
if((1<<i)& dc)
u=fa[u][i];
}
if(u==v) //如果深度一样时,两个点相同,直接返回
return u;
for(i=16;i>=0;i--) //如果从小到大的话就有可能无法提到正确位置
{
if(fa[u][i]!=fa[v][i])//跳2^j步,如果不一样,就跳,否则不跳 如果相等有可能都为0 即超过根节点
{//在同一高度了,他们一起向上跳,跳他们不相同节点,当全都跳完之后fa[u][0]就是lca
u=fa[u][i];
v=fa[v][i];
}
}
u=fa[u][0]; //上述过程做完,两点都在LCA下一层,所以走一步即可
return u;
}
int main()
{
int t;
int x,y,z;
scanf("%d",&t);
for(int k=1;k<=t;k++)
{
init();
scanf("%d%d",&n,&m);
for(int i=0;i<=n;i++)
g[i].clear();
for(int i=1;i<n;i++)
{
scanf("%d%d%d",&x,&y,&z);
g[x].push_back(node(y,z));
fa[y][0]=x;
if(fa[x][0]==0) //查找根节点
root=x;
}
d[root]=1;
dfs(root);
bz();
int a,b;
for(int i=0;i<m;i++)
{
scanf("%d%d",&a,&b);
int ans=dis[a]+dis[b]-2*dis[lca(a,b)];
printf("%d\n",ans);
}
}
return 0;
}