hdu 2586 How far away ？ 倍增求LCA

倍增求LCA

LCA函数返回(u,v)两点的最近公共祖先

#include <bits/stdc++.h>
using namespace std;

const int N = *;
struct node {
    int v,val,next;
    node(){}
    node(int vv,int va,int nn):v(vv),val(va),next(nn){}
}E[N];
int n,m;
int tot,head[N],dis[N],f[N][],dep[N];

void init() {
    tot = ;
    memset(head,,sizeof(head));
    memset(dis,,sizeof(dis));
    memset(f,,sizeof(f));
    memset(dep,,sizeof(dep));
}

void add(int u,int v,int val) {
    E[++tot].next = head[u];
    E[tot].v = v;
    E[tot].val = val;
    head[u] = tot;
}

void addEdge(int u,int v,int val) {
    add(u,v,val);
    add(v,u,val);
}

void dfs(int x,int fa) {
    f[x][] = fa;
    for(int i=;f[x][i-];i++)
        f[x][i] = f[f[x][i-]][i-];
    for(int i=head[x]; i; i=E[i].next) {
        int v = E[i].v;
        if(v != fa) {
            dis[v] = dis[x] + E[i].val;
            dep[v] = dep[x] + ;
            dfs(v,x);
        }
    }
}

int lca(int u,int v) {
    if(dep[u] < dep[v]) swap(u,v);
    //int ans = dep[u] - dep[v];
    for(int i=;i>=;i--)
        if(dep[u]-(<<i)>=dep[v])
            u = f[u][i];
    for(int i=;i>=;i--) {
        if(f[u][i]!=f[v][i])
            u=f[u][i],v=f[v][i];//ans+=(2<<i);
    }
    //if(u!=v) ans+=2;
    //return ans;
    if(u!=v) return f[u][];
    else return u;
}

int main () {
    //freopen("in.txt","r",stdin);
    int T; scanf("%d",&T);
    while (T--) {
        init();
        scanf("%d %d",&n, &m);
        for(int i=;i<=n-;i++) {
            int u,v,val;
            scanf("%d %d %d",&u,&v,&val);
            addEdge(u,v,val);
        }
        dfs(,);
        // cout <<"yes" <<endl;
        while(m--) {
            int u,v; scanf("%d %d",&u,&v);
            int  t = lca(u,v);
            printf("%d\n",dis[u]+dis[v]-*dis[t]);
        }
    }
    return ;
}