最近公共祖先【模板题】【树上倍增】【HDU2586】

题目传送门

树上俩点之间的距离
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;
}