BZOJ 1602: [Usaco2008 Oct]牧场行走 倍增裸题

Description

N头牛（2<=n<=1000）别人被标记为1到n，在同样被标记1到n的n块土地上吃草，第i头牛在第i块牧场吃草。 这n块土地被n-1条边连接。 奶牛可以在边上行走，第i条边连接第Ai，Bi块牧场，第i条边的长度是Li（1<=Li<=10000）。 这些边被安排成任意两头奶牛都可以通过这些边到达的情况，所以说这是一棵树。 这些奶牛是非常喜欢交际的，经常会去互相访问,他们想让你去帮助他们计算Q(1<=q<=1000)对奶牛之间的距离。

Input

*第一行：两个被空格隔开的整数：N和Q

*第二行到第n行：第i+1行有两个被空格隔开的整数：AI，BI，LI

*第n+1行到n+Q行：每一行有两个空格隔开的整数：P1，P2，表示两头奶牛的编号。

Output

*第1行到第Q行：每行输出一个数，表示那两头奶牛之间的距离。

题解：

刷水有益健康.

Code:

#include <bits/stdc++.h>
#define maxn 10000
#define setIO(s) freopen(s".in","r",stdin)
using namespace std;
int hd[maxn],to[maxn],nex[maxn],val[maxn],dep[maxn],f[23][maxn],len[maxn];
int cnt;
void add(int u,int v,int c){
    nex[++cnt]=hd[u],hd[u]=cnt,to[cnt]=v,val[cnt]=c;
}
void dfs(int u,int fa){

    f[0][u]=fa;
    for(int i=1;i<=20;++i) f[i][u]=f[i-1][f[i-1][u]];
    for(int i=hd[u];i;i=nex[i]) {
        if(to[i]==fa) continue;
        len[to[i]]=len[u]+1, dep[to[i]]=dep[u]+val[i],dfs(to[i],u);
    }
}
int lca(int a,int b){
    if(len[a]>len[b]) swap(a,b);
    if(len[a]!=len[b]){
        for(int i=20;i>=0;--i) if(len[f[i][b]] >= len[a]) b=f[i][b];
    }
    if(a==b) return a;
    for(int i=20;i>=0;--i) {
        if(f[i][a]!=f[i][b]) a=f[i][a],b=f[i][b];
    }
    return f[0][a];
}
int dis(int a,int b){
    return dep[a]+dep[b]-(dep[lca(a,b)]<<1);
}
int main(){
    // setIO("input");
    int n,q;
    scanf("%d%d",&n,&q);
    for(int i=1,a,b,c;i<n;++i){
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c),add(b,a,c);
    }
    dfs(1,0);
    while(q--){
        int a,b;
        scanf("%d%d",&a,&b);
        printf("%d\n",dis(a,b));
    }
    return 0;
}