Codevs 2370 小机房的树

题目描述 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

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

const int maxx = 50000 + 100;

int head[maxx],deep[maxx];//shendu
int anc[maxx][20+2],dis[maxx][20+2];//zuxian   yugenjiediandejvli
bool done[maxx];
int n,m,x,y,z,num,root;

struct Edge{
    int next;//qidian
    int to;//zhongdian 
    int value;//jingli
}Edges[maxx<<1];

void build(int x,int y,int z){     //建边 
    Edges[++num].to=y;
    Edges[num].next=head[x];

    //cout<<head[x]<<"   "; 

    Edges[num].value=z;
    head[x]=num;

    //cout<<num<<"   ";
}

void Dfs(int x){
    done[x]=true;
    for(int i=1;i<=20;i++){
        if((1<<i) > deep[x]) break;
        anc[x][i]=anc[anc[x][i-1]][i-1];
        dis[x][i]=dis[x][i-1]+dis[anc[x][i-1]][i-1];
    }
    for(int i=head[x];i;i=Edges[i].next){
        int now=Edges[i].to;
        if(done[now]) continue;
        deep[now]=deep[x]+1;
        anc[now][0]=x;
        dis[now][0]=Edges[i].value;
        Dfs(now);
    }
}

inline int Lca(int x,int y){
    int Ans=0;
    if(deep[x]>deep[y]) swap(x,y);
    int d=deep[y]-deep[x];//d:深度差 
    for(int i=0;i<=20;i++){

        if((1<<i) & d) {                     //........................

           Ans += dis[y][i];
           y=anc[y][i];
        }
    }
    for(int i=20;i>=0;i--){
        if(anc[x][i]!=anc[y][i]){
            Ans += dis[x][i];
            Ans += dis[y][i];
            x=anc[x][i];
            y=anc[y][i];
        }
    }
    if(x==y) return Ans;
    else{
        Ans+=dis[x][0]+dis[y][0];
        return Ans;
    }
}

int main(){
    cin>>n;
    for(int i=1;i<n;i++){
        cin>>x>>y>>z;//qidian zhongdian jingli 
        build(x,y,z);   build(y,x,z);//建双向边 
    }
    Dfs(0);//初始化 ,以0为根节点 
    cin>>m;
    while( m-- ){//多组询问 
        cin>>x>>y;
        printf("%d\n",Lca(x,y));
    }
    return 0;
}