树上倍增 hdu 2586

参考博客：

代码：

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<string.h>
using namespace std;
const int max_=8e4+;
int depth[max_];//节点深度
int gw[max_][];//第 i 点到2^j父亲的距离
int fa[max_][];//第 i 点的2^j父亲
bool vis[max_];//标记数组
int tot,N;
struct Tree//树
{
     int to;
     int w;
     int next;
}a[max_*];
int edge[max_*];
void add_edge(int x,int y,int w)//建树
{
    a[tot].to=y;
    a[tot].w=w;
    a[tot].next=edge[x];
    edge[x]=tot++;
}
void dfs(int now,int deep)//dfs预处理
{
    if(vis[now]==)
    {
        vis[now]=;
        depth[now]=deep;//记录节点的深度
    }
    for(int i=edge[now];i!=-;i=a[i].next)
    {
        int to=a[i].to;
        if(vis[to])
        continue;
        int w=a[i].w;
        gw[to][]=w;//to到父亲的距离
        fa[to][]=now;//to的父亲节点
        dfs(to,deep+);
    }
}
void LCA_init(int n)//预处理
{
    for(int i=;i<=n;i++)
        for(int j=;j<=N;j++)
    {
        fa[i][j]=fa[fa[i][j-]][j-];
        gw[i][j]=gw[fa[i][j-]][j-]+gw[i][j-];
    }
}
int LCA_Q(int u,int v)
{
    int ans=;//距离
    if(depth[u]<depth[v])
        swap(u,v);//保证大减小
    int k;
        k=depth[u]-depth[v];
    for(int i=;(<<i)<=k;i++)
    {
        if((<<i)&k)
           ans+=gw[u][i],u=fa[u][i];
    }
        if(u==v)
            return ans;//找到了就退出
        for(int i=N;i>=;i--)
        {
            if(fa[u][i]!=fa[v][i])
            {
                ans+=gw[u][i],
                ans+=gw[v][i];
                u=fa[u][i],
                v=fa[v][i];
            }
        }
        return ans+=gw[v][]+gw[u][];
}
void init(int n)//初始化
{
    memset(vis,,sizeof(vis));
    tot=;
    N=(int)(log(n+0.0)/log());
    memset(edge,-,sizeof(edge));
}
int main()
{
    int t;
    cin>>t;
    while(t--)
    {
        int n,m;
        scanf("%d %d",&n,&m);
        init(n);
        for(int i=;i<n-;i++)
        {
            int u,v,w;
            scanf("%d %d %d",&u,&v,&w);
            add_edge(u,v,w);
            add_edge(v,u,w);
        }
        dfs(,);
        LCA_init(n);
        while(m--)
        {
            int u,v;
            scanf("%d %d",&u,&v);
            int L=LCA_Q(u,v);
            printf("%d\n",L);
        }
       /* for(int i=1;i<=n;i++)
            printf("%d ",fa[i][2]);*/
    }
}
/*
13 2
1 2 1
1 3 1
1 4 1
2 5 1
3 6 1
6 10 1
6 11 1
10 13 1
4 7 1
4 8 1
4 9 1
8 12 1
*/
  /*  for(int x=max0;x>=0;x--)     //注意！此处循环必须是从大到小!因为我们应该越提越“精确”，
        if(fa[u][x]!=fa[v][x])   //如果从小到大的话就有可能无法提到正确位置，自己可以多想一下
        {
            u=fa[u][x];
            v=fa[v][x];
        }
    return fa[u][0];    //此时u、v的第一个父亲就是LCA。*/