Hdu 2586 lca入门

题意：给一个树，多次询问两点之间距离。
做法：就是两点到根距离和减去两倍lca到根距离。主要是求lca，这里参考了这篇文章，http://wenku.baidu.com/view/392c7023dd36a32d73758193.html，把lca转化成rmq问题用st表搞定。

代码：

#include <bits/stdc++.h>
using namespace std;
vector<pair<int, int> >g[40500];
int n, q;
bool vis[40500];
int fi[40500];
int f[80500][20];
int d[80500];
int rev[80500];
int l[40500];
int cnt;
void dfs(int st, int flor, int dis){
    vis[st]=true;
    l[st]=dis;
    for(vector<pair<int, int> >::iterator it=g[st].begin();it!=g[st].end();it++){
        if(vis[(*it).first])continue;
        ++cnt;
        if(fi[(*it).first]==0)fi[(*it).first]=cnt;
        d[cnt]=flor+1;
        rev[cnt]=(*it).first;
        dfs((*it).first, flor+1, dis+(*it).second);
        ++cnt;
        d[cnt]=flor;
        rev[cnt]=st;
    }
}

void ST()
{
    for(int i=1;i<=cnt;i++)f[i][0]=i;
    for(int j=1;j<20;j++)
        for(int i=1;i<=cnt;i++){
            if(i+(1<<j)-1>cnt)break;
            int a=f[i][j-1];
            int b=f[i+(1<<(j-1))][j-1];
            if(d[a]>d[b])f[i][j]=b;
            else f[i][j]=a;
        }
}

int rmq(int x, int y){
    int j=31-__builtin_clz(y-x+1);
    int a=f[x][j];
    int b=f[y-(1<<j)+1][j];
    if(d[a]>d[b])return rev[b];
    else return rev[a];
}

int main()
{
    int t;
    scanf("%d", &t);
    for(int tt=1;tt<=t;tt++){
        memset(vis, false, sizeof(vis));
        memset(fi, 0, sizeof(fi));
        memset(f, 0, sizeof(f));
        memset(l, 0, sizeof(l));
        scanf("%d%d", &n, &q);
        for(int i=1;i<=n;i++)g[i].clear();
        for(int i=1;i<n;i++){
            int x, y, l;
            scanf("%d%d%d", &x, &y, &l);
            g[x].push_back(make_pair(y, l));
            g[y].push_back(make_pair(x, l));
        }
        cnt=1;
        fi[1]=1;
        d[1]=0;
        rev[1]=1;
        dfs(1, 0, 0);
        ST();
        for(int i=1;i<=q;i++){
            int x, y;
            scanf("%d%d", &x, &y);
            int ans=l[x]+l[y];
            x=fi[x];
            y=fi[y];
            if(x>y)swap(x, y);
            int lca=rmq(x, y);
            ans-=2*l[lca];
            printf("%d\n", ans);
        }
    }
}