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

题解：调了一下午。伤心透顶。然后HWH大佬从我身边经过，一眼指出我最后输出的时候dis没有乘以2。QAQ。我真是一个辣鸡。

 

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#define N 50010
using namespace std;
int n,m,num(0);
int head[N],deep[N],dis[N]={0};
int f[N][25];
struct node
{
    int v,t,pre;
}e[2*N];
void add(int from,int to,int ti)
{
    e[++num].v=to;
    e[num].t=ti;
    e[num].pre=head[from];
    head[from]=num;
}
void dfs(int ki,int dep)//查找每个点的深度和与根节点的距离 
{
    deep[ki]=dep;
    for (int i=head[ki];i;i=e[i].pre)
      {
           int u=e[i].v;
           if (!deep[u]&&u)
             {
                   f[u][0]=ki;
                   dis[u]=dis[ki]+e[i].t;
                   dfs(u,dep+1);
           }
      }
}
void get_father()//倍增找父节点 
{
    for (int j=1;j<=21;j++)
      for (int i=0;i<n;i++)
        f[i][j]=f[f[i][j-1]][j-1];
}
int lca(int x,int y)
{
    for (int j=21;j>=0;j--)//将深度较深的点拉到深度比较浅的点的那一层 
      if (deep[f[x][j]]>=deep[y]) x=f[x][j];
    for (int j=21;j>=0;j--)//找最近公共祖先 
      if (f[x][j]!=f[y][j])
        {
            x=f[x][j];
            y=f[y][j];
        }
    if (x==y) return x;//两个点在一条链上 
    return f[x][0];
}
int main()
{
    scanf("%d",&n);
    for (int i=1,u,v,c;i<n;i++)
      {
           scanf("%d%d%d",&u,&v,&c);
           add(u,v,c);
           add(v,u,c);
      }
    dfs(0,0);
    get_father();
    scanf("%d",&m);
    for (int i=1,u,v,t;i<=m;i++)
      {
           scanf("%d%d",&u,&v);
           if (deep[u]<deep[v]) swap(u,v); 
         int Fa=lca(u,v);
         printf("%d\n",dis[u]+dis[v]-2*dis[Fa]);
      }
    return 0;
}