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 50005int 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));    }}