2370 小机房的树 （倍增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模板

#include<cstdio>
#include<iostream>
#define M 1000001
using namespace std;
int f[M][31];
int dis[M];
struct node {
    int to,next;
    int val;
};
node e[M];
int tot,head[M];
int n,m;
int deep[M];
inline int read(int&x) {
    x=0;char c=getchar();
    while(c>'9'||c<'0') c=getchar();
    while(c>='0'&&c<='9') x=10*x+c-48,c=getchar();
}
inline void dfs(int now,int from,int dep,int v) {
    deep[now]=dep;
    f[now][0]=from;
    dis[now]=v+dis[from];
    for(int i=head[now];i;i=e[i].next) {
        int u=e[i].to;
        if(u!=from) dfs(u,now,dep+1,e[i].val); 
    }
}
/*inline void dfs(int u,int dep) {//两种dfs姿势都可以
    deep[u]=dep;
    for(int i=head[u];i;i=e[i].next) {
        int v=e[i].to;
        if(!deep[v]&&v) {
            f[v][0]=u;
            dis[v]=dis[u]+e[i].val;
            dfs(v,dep+1);
        }
    }
}*/
inline void add(int x,int y,int z) {
    e[++tot].to=y;
    e[tot].val=z;
    e[tot].next=head[x];
    head[x]=tot;
}
inline int LCA(int a,int b) {
    int ans=0;
    if(deep[a]<deep[b]) swap(a,b);
    int t=deep[a]-deep[b];
    for(int i=0;i<=30;i++) 
      if((1<<i)&t) a=f[a][i];
    if(a==b) return a;
    for(int i=30;i>=0;i--) 
      if(f[a][i]!=f[b][i]) 
        a=f[a][i],b=f[b][i];
    return f[a][0];
}
int main() {
    read(n); 
    for(int i=1;i<n;i++) {
        int x,y,z;
        read(x);read(y);read(z);
        x++;y++;
        add(x,y,z);add(y,x,z);
    }
    dfs(1,1,0,0);
    for(int j=1;j<=30;j++)
      for(int i=1;i<=n;i++)
        f[i][j]=f[f[i][j-1]][j-1];
    read(m);
    for(int i=1;i<=m;i++) {
        int x,y;
        read(x);read(y);
        x++;y++;
        int lca=LCA(x,y);
        if(lca==x) printf("%d\n",dis[y]-dis[x]);
        else printf("%d\n",dis[x]+dis[y]-2*dis[lca]);
    }
    return 0;
}