小机房的树

小机房的树

算法使用：
LCA
题目描述：
小机房有棵焕狗种的树，树上有N个节点，节点标号为0到N-1，有两只虫子名叫飘狗和大吉狗，分居在两个不同的节点上。有一天，他们想爬到一个节点上去搞基，但是作为两只虫子，他们不想花费太多精力。已知从某个节点爬到其父亲节点要花费 c 的能量（从父亲节点爬到此节点也相同），他们想找出一条花费精力最短的路，以使得搞基的时候精力旺盛，他们找到你要你设计一个程序来找到这条路，要求你告诉他们最少需要花费多少精力
输入描述 Input Description
第一行一个n，接下来n-1行每一行有三个整数u，v, c 。表示节点 u 爬到节点 v 需要花费 c 的精力。
第n+1行有一个整数m表示有m次询问。接下来m行每一行有两个整数 u ，v 表示两只虫子所在的节点
输出描述：
一共有m行，每一行一个整数，表示对于该次询问所得出的最短距离。
样例输入：
3
1 0 1
2 0 1
3
1 0
2 0
1 2
样例输出：
1
1
2
数据范围及提示：
1<=n<=50000， 1<=m<=75000， 0<=c<=1000

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=50010;
struct node
{
    int to;
    int w;
    int next;
}e[maxn*10];
int n,m,tot,dis[maxn][25],deep[maxn],f[maxn][30],head[maxn];
void edd_edge(int u,int v,int w)
{
    tot++;
    e[tot].to=v;
    e[tot].w=w;
    e[tot].next=head[u];
    head[u]=tot;
}
int build(int u)
{
    for(int i=head[u];i;i=e[i].next)
    if(!f[e[i].to][0])
    {
        f[e[i].to][0]=u;
        deep[e[i].to]=deep[u]+1;
        dis[e[i].to][0]=e[i].w;
        build(e[i].to);
    }
}
void init()
{
    for(int j=1;j<=20;j++)
      for(int i=1;i<=n;i++)
      {
        f[i][j]=f[f[i][j-1]][j-1];
        dis[i][j]=dis[i][j-1]+dis[f[i][j-1]][j-1];
      }
}
int lca(int a,int b)
{
    int ans=0;
    if(deep[a]<deep[b])
    swap(a,b);
    if(deep[a]!=deep[b])
    {
        int d=deep[a]-deep[b];
        for(int i=0;d;i++)
        {
            if(d&1)
            {
                ans+=dis[a][i];
                a=f[a][i];
            }
            d>>=1;
        }
    }
    if(a==b)
    return ans;
    for(int i=20;i>=0;i--)
    if(f[a][i]!=f[b][i])
    {
        ans+=dis[a][i];
        ans+=dis[b][i];
        a=f[a][i];
        b=f[b][i];
    }
    ans+=dis[a][0];
    ans+=dis[b][0];
    return ans;
}
int main()
{
    int x,y,z;
    scanf("%d",&n);
    for(int i=1;i<=n-1;i++)
    {
        scanf("%d%d%d",&x,&y,&z);
        edd_edge(x+1,y+1,z);
        edd_edge(y+1,x+1,z);
    }
    f[1][0]=1;
    build(1);
    init();
    int start,end;
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&start,&end);
        int tmp=lca(start+1,end+1);
        printf("%d
",tmp);
    }
    return 0;
}