codevs 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

思路

很明显，n-1条边就是树了。他要求两个点之间的最小路径长，想到LCA。
与查询不同的是，在dfs时不仅仅更新deep【】，还要更新dis【】与fa【】。
花了两次才AC这道题。。。第一次WA的时候发现数据剧毒。。。然后发现是读入写错了？！？我rmp的。。。

代码

这不清真

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
const int N=60000+5;
const int P=20+1;
int n,m,head[N],num=0;
int fa[N],dis[N],flag[N],deep[N],anc[N][P+1];
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
struct edge
{
    int u,v,w;
    int next;
}ed[2*N];
void build(int u,int v,int w)
{
    ed[++num].v=v;
    ed[num].u=u;
    ed[num].w=w;
    ed[num].next=head[u];
    head[u]=num;
}
void dfs(int u)
{
    anc[u][0]=fa[u];
    for (int i=1;i<=20;i++)
    anc[u][i]=anc[anc[u][i-1]][i-1];
    for (int i=head[u];i!=-1;i=ed[i].next)
    {
        int v=ed[i].v;
        if(!flag[v])
        {
            flag[v]=1;
            deep[v]=deep[u]+1;
            dis[v]=ed[i].w+dis[u];
            fa[v]=u;
            dfs(v);
        }
    }
}
int LCA(int x,int y)
{
    int a=x,b=y;
    if (deep[x]<deep[y]) {swap(a,b);swap(x,y);}
    for (int i=20;i>=0;i--)
    if (deep[y]<=deep[anc[x][i]]) x=anc[x][i];
    if (x==y) return dis[a]-dis[x];
    for (int i=20;i>=0;i--)
    if (anc[x][i]!=anc[y][i]) {x=anc[x][i];y=anc[y][i];}
    return dis[a]-dis[anc[x][0]]*2+dis[b];
}
void clear()
{
    memset(head,-1,sizeof(head));
    memset(anc,-1,sizeof(anc));
    memset(dis,0,sizeof(dis));
    memset(deep,0,sizeof(deep));
    memset(flag,0,sizeof(flag));
}
int main()
{
    n=read();
    clear();
    for (int i=1;i<n;i++)
    {
        int u=read(),v=read(),w=read();
        build(u,v,w);
        build(v,u,w);
    }
    m=read();
    deep[0]=0;flag[0]=1;
    dfs(0);
    for (int i=1;i<=m;i++)
    {
        int x=read(),y=read();
        printf("%d\n",LCA(x,y));
    }
    return 0;
}