hdu 2586 How far away ？倍增LCA

hdu 2586 How far away ？倍增LCA

题目链接

http://acm.hdu.edu.cn/showproblem.php?pid=2586

思路：

• 针对询问次数多的时候，采取倍增求取LCA，同时跟新距离数组
• 因为
• \(2^{16} > 40000\)
• 所以所以表示祖先的数组dp[][]第二维取到16即可
• 就这道题来说，与比较tarjan比较，稍快一点

代码：

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <string.h>
#include <math.h>
using namespace std;
const int maxn = 40005;
const int maxm = 80005;
struct node {
    int to,next,w;
}edges[maxm];
int head[maxn],cnt,dp[maxn][17],dep[maxn],dist[maxn];
inline void addedge(int u, int v, int w) {
    edges[cnt].to=v;
    edges[cnt].w=w;
    edges[cnt].next=head[u];
    head[u]=cnt++;
}
void dfs(int s, int x) {
    dp[s][0]=x;
    dep[s]=dep[x]+1;
    int t;
    for(int i=1;(1<<i)<=dep[s];++i)
        dp[s][i]=dp[dp[s][i-1]][i-1];
    for(int i=head[s];i!=-1;i=edges[i].next) {
        t=edges[i].to;
        if(t==x) continue;
        dist[t]=dist[s]+edges[i].w;
        dfs(t,s);
    }
}
inline int lca(int u, int v) {
    if(dep[v]>dep[u]) swap(u,v);
    for(int i=16;i>=0;--i) {
        if((1<<i)<=(dep[u]-dep[v])) {
            u=dp[u][i];
        }
    }
    if(u==v) return u;
    for(int i=16;i>=0;--i) {
        if((1<<i)<=dep[u]&&(dp[v][i]!=dp[u][i])) {
            u=dp[u][i];
            v=dp[v][i];
        }
    }
    return dp[u][0];
}
inline int slove(int u ,int v) {
    int z=lca(u,v);
    return dist[u]-2*dist[z]+dist[v];
}
inline void init() {
    cnt=0;
    memset(head,-1,sizeof(head));
}
int main() {
    int t,n,m,u,v,w;
    scanf("%d",&t);
    while(t--) {
        scanf("%d %d",&n,&m);
        init();
        for(int i=1;i<n;++i) {
            scanf("%d %d %d",&u,&v,&w);
            addedge(u,v,w);
            addedge(v,u,w);
        }
        dep[1]=0;//保持dfs的统一，实际dep[1]=1
        dist[1]=0;
        dfs(1,1);
        for(int i=1;i<=m;++i) {
            scanf("%d %d",&u,&v);
            printf("%d\n",slove(u,v));
        }
    }
    return 0;
}