码迷,mamicode.com
首页 > 其他好文 > 详细

放一道比较基础的LCA 的题目把 :CODEVS 2370 小机房的树

时间:2017-10-27 15:29:20      阅读:126      评论:0      收藏:0      [点我收藏+]

标签:clu   names   长度   --   节点   找到你   需要   desc   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

 

注意如何描述长度!

先来一发倍增的

#include<cstdio>
#include<algorithm>
#include<cmath>
#define maxn 1000500
#define ll long long
using namespace std;
struct st{
    ll u,v,val,next;
}s[maxn];
ll head[maxn],tot,n,m,u,v,val,ceng[maxn],ju[maxn],fa[maxn][20];

inline void add(ll u,ll v,ll val)
{
    ++tot;
    s[tot].u = u;
    s[tot].v = v;
    s[tot].val = val;
    s[tot].next = head[u];
    head[u] = tot;
}

inline void dfs(ll f,ll now)
{
    fa[now][0] = f;
    ceng[now] = ceng[f] + 1;
    for(ll i=head[now];i;i=s[i].next)
    {
        v = s[i].v;
        if(v != f)
        {
            ju[v] = ju[now] + s[i].val;//没错!就是这里!ju[i]存着点i到根节点的距离!!! 
            dfs(now , v);
        }
    }
}

inline void doit()
{
    for(ll j = 1;(1 << j) <= n;j++)
        for(ll i=1;i<=n;i++)
            fa[i][j] = fa[fa[i][j- 1]][j -1];
}

inline ll lca(ll a,ll b)
{
    if(ceng[a] > ceng[b]) swap(a,b);
    ll f = ceng[b] - ceng[a];
    for(ll i = 0;(1 << i) <= n;i++)
        if((1 << i) & f)
            b = fa[b][i];
    if(a != b)
    {
        for(ll i=(ll)log2(n);i>=0;i--)
        {
            if(fa[a][i] != fa[b][i])
            {
                a = fa[a][i];
                b = fa[b][i];
            }
        }
        a = fa[a][0];
    }
    return a;
}

int main(){
    scanf("%lld",&n);
    
    for(ll i=1;i<n;i++)
    {
        scanf("%lld%lld%lld",&u,&v,&val);
        u++;//我不喜欢有0的情况,所以我就都加一了!!! 
        v++;
        add(u,v,val);
        add(v,u,val);
    }
    
    dfs(0,1);
    
    doit();
    
    scanf("%lld",&m);
    
    for(ll i=1;i<=m;i++)
    {
        scanf("%lld%lld",&u,&v);
        u++;
        v++;
        ll l = lca(u,v);
        printf("%lld\n",ju[u] + ju[v] - ju[l] * 2);//求出两点之间距离(自己想想为什么。。。) 
    }
}

 

 

 

放一道比较基础的LCA 的题目把 :CODEVS 2370 小机房的树

标签:clu   names   长度   --   节点   找到你   需要   desc   lca   

原文地址:http://www.cnblogs.com/kczno1fans/p/7742853.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!