Codevs_P2370 小机房的树(LCA)

题目描述 Description
小机房有棵焕狗种的树，树上有N个节点，节点标号为0到N-1，有两只虫子名叫飘狗和大吉狗，分居在两个不同的节点上。有一天，他们想爬到一个节点上去搞基，但是作为两只虫子，他们不想花费太多精力。已知从某个节点爬到其父亲节点要花费 c 的能量（从父亲节点爬到此节点也相同），他们想找出一条花费精力最短的路，以使得搞基的时候精力旺盛，他们找到你要你设计一个程序来找到这条路，要求你告诉他们最少需要花费多少精力

输入描述 Input Description
第一行一个n，接下来n-1行每一行有三个整数u，v, c 。表示节点 u 爬到节点 v 需要花费 c 的精力。
第n+1行有一个整数m表示有m次询问。接下来m行每一行有两个整数 u ，v 表示两只虫子所在的节点
输出描述 Output Description
一共有m行，每一行一个整数，表示对于该次询问所得出的最短距离。

样例输入 Sample Input
3
1 0 1
2 0 1
3
1 0
2 0
1 2

样例输出 Sample Output
1
1
2

数据范围及提示 Data Size & Hint
1<=n<=50000， 1<=m<=75000， 0<=c<=1000

思路:很裸的LCA，但比较麻烦，先DFS计算深度，在处理它2^i的数据,需要两个数组存跳2^i步的位置及权值，存算跳2^i步的时候忘记加上前面走过的了，调了半天qwq
PS:出题人脑洞有点大23333333

#include<cstdio>
#include<vector>
#include<algorithm>
#include<iostream>
using namespace std;
#define N 50005
int n,m;
int deep[N];
int f[N][31],t[N][31];
struct g{
    vector<int> q;
    vector<int> v;
}w[N];
void dfs(int x,int d){
    deep[x]=d;
    for(int i=0;i<w[x].q.size();i++)
        if(!deep[w[x].q[i]]){
            dfs(w[x].q[i],d+1);
            f[w[x].q[i]][0]=x;
            t[w[x].q[i]][0]=w[x].v[i];
        }
}
void work(){
    for(int i=1;i<=30;i++)
        for(int j=0;j<n;j++){
            f[j][i]=f[f[j][i-1]][i-1];
            t[j][i]=t[j][i-1]+t[f[j][i-1]][i-1];
        }
}
int lca(int u,int v){
    if(deep[u]<deep[v]) swap(u,v);
    int l=deep[u]-deep[v],vv=0;
    if(l)
        for(int i=0;i<=30;i++){
            if((1<<i)&l) vv+=t[u][i],u=f[u][i];
        }
    if(u==v) return vv;
    for(int i=30;i>=0;i--){
        if(f[u][i]!=f[v][i]){
            vv+=t[u][i]+t[v][i];
            u=f[u][i];v=f[v][i];
        }
    }
    vv+=t[u][0]+t[v][0];
    return vv;
}
int main(){
    scanf("%d",&n);int u,v,c;
    for(int i=1;i<n;i++){
        scanf("%d%d%d",&u,&v,&c);
        w[u].q.push_back(v);w[u].v.push_back(c);
        w[v].q.push_back(u);w[v].v.push_back(c);
    }
    dfs(0,1);work();
    scanf("%d",&m);
    while(m--){
        scanf("%d%d",&u,&v);
        printf("%d\n",lca(u,v));
    }
}