小机房的树 codevs 2370

2370 小机房的树

时间限制: 1 s

空间限制: 256000 KB

题目等级 : 钻石 Diamond

题解

查看运行结果

题目描述 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强行ac（极端数据会超时，但是我还是ac了）

来，上代码：

#include<cstdio>
#include<algorithm>

using namespace std;

struct node {
    int to,dis,next;
};
struct node edge[50000*2+1];

int n,m,head[50001],num=0,dfn[50001],flag=0;

char ch;

inline void read_int(int &x)
{
    x=0;ch=getchar();
    while(ch>‘9‘||ch<‘0‘) ch=getchar();
    while(ch>=‘0‘&&ch<=‘9‘){x=x*10+(int)(ch-‘0‘);ch=getchar();}
}

inline void edge_add(int from,int to,int dis)
{
    num++;
    edge[num].to=to;
    edge[num].dis=dis;
    edge[num].next=head[from];
    head[from]=num;
}

void dfs(int now)
{
    dfn[now]=++flag;
    for(int i=head[now];i;i=edge[i].next) if(!dfn[edge[i].to]) dfs(edge[i].to);
}

int lca(int now_1,int now_2)
{
    int dis_1=0,dis_2=0;
    while(dfn[now_2]>dfn[now_1])
    {
        for(int i=head[now_2];i;i=edge[i].next)
        {
            if(dfn[edge[i].to]<dfn[now_2])
            {
                now_2=edge[i].to;
                dis_2+=edge[i].dis;
            }
        }
    }
    while(dfn[now_1]>dfn[now_2])
    {
        for(int i=head[now_1];i;i=edge[i].next)
        {
            if(dfn[edge[i].to]<dfn[now_1])
            {
                now_1=edge[i].to;
                dis_1+=edge[i].dis;
            }
        }
    }
    return dis_1+dis_2;
}

int main()
{
    read_int(n);
    int from,to,dis;
    for(int i=1;i<n;i++)
    {
        read_int(from),read_int(to),read_int(dis);
        edge_add(from,to,dis);
        edge_add(to,from,dis);
    }
    read_int(m);
    dfs(1);
    for(int i=1;i<=m;i++)
    {
        read_int(from),read_int(to);
        if(dfn[from]>dfn[to]) swap(from,to);
        printf("%d\n",lca(from,to));
    }
    return 0;
}