codevs 2370 小机房的树

Posted 2020-08-19 惑&葬

题目描述 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<cstring>
#include<iostream>
#define maxn 500001
using namespace std;
int n,m,s,tot,deep[maxn],head[maxn],fa[maxn][21],ans,v[maxn];
struct node{
    int to,next,w;
}a[maxn*3];
void add(int x,int y,int z)
{
    tot++;
    a[tot].to=y;
    a[tot].next=head[x];
    a[tot].w=z;
    head[x]=tot;
}
void dfs(int x,int y,int d)
{
    fa[x][0]=y;
    deep[x]=d;
    for(int i=head[x];i;i=a[i].next)
      if(y!=a[i].to)
        v[a[i].to]=v[x]+a[i].w,dfs(a[i].to,x,d+1);
}
int lca(int x,int y)
{
    int i,j;
    if(deep[x]<deep[y])
      swap(x,y);
    int t=deep[x]-deep[y];
    for(i=0;i<=20;i++)
      if(t&(1<<i)) 
        x=fa[x][i];
    if(x==y) 
      return x;
    for(j=20;j>=0;j--)
      if(fa[x][j]!=fa[y][j])                                    
        x=fa[x][j],y=fa[y][j];
    return fa[x][0];
}
int main()
{
    int i,j,x,y,z;
    scanf("%d",&n);
    for(i=1;i<n;i++)
      scanf("%d%d%d",&x,&y,&z),add(x,y,z),add(y,x,z);
    dfs(0,0,0);
    for(j=1;j<=20;j++)
      for(i=1;i<=n;i++)
        fa[i][j]=fa[fa[i][j-1]][j-1];
    scanf("%d",&m);
    for(i=1;i<=m;i++)
    {
        scanf("%d%d",&x,&y),s=lca(x,y);
        if(s==y)
           printf("%d\n",v[x]-v[y]);
        else
          printf("%d\n",v[x]+v[y]-2*v[s]);
    }
      
    return 0;
}