CODE[VS] 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

题目说的很详细 求两个点距离父节点的最短距离 也就是求它们的LCA

思想:
两个深度不同的点要找他们的LCA 我们需要让他们先达到同一深度在一起向上跑 知道在第一个相同的点会合
首先 我们需要构建这棵树 用邻接表存图解决 同时记录一下每个点的深度和父节点(每个点从某个点可以访问到 那么这个点就是父节点)子节点的深度 == 父节点深度 + 1
其次 我们就可以进行LCA的过程了 我们默认其中一个点的深度 > 另一个点的深度
我们让深度更大的那个点先向上一直走 更新答案 直到两者深度相同 同时更新当前移动的点为它的父节点
当处于同一深度之后 就可以让两个点一起向上跑 更新答案 同时更新它们所处的位置也就是各自的父节点
注意树根的深度为一

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
const int MAXN = 50000 + 50;
int n, tot = 0, m;
int first[MAXN], nxt[MAXN << 1], dis[MAXN], deep[MAXN], fa[MAXN];
bool vis[MAXN];
struct edge{
    int from, to, cost;
}es[MAXN << 1];
void build(int ff, int tt, int dd)
{
    es[++tot] = (edge){ff,tt,dd};
    nxt[tot] = first[ff];
    first[ff] = tot;
}
void dfs(int x)
{
    for(int i = first[x]; i != -1; i = nxt[i])
    {
        int v = es[i].to;
        if(!vis[v])
        {
            vis[v] = 1;
            dis[v] = es[i].cost;
            deep[v] = deep[x] + 1;
            fa[v] = x;
            dfs(v);
        }
    }
}
int lca(int x, int y)
{
    int ans = 0;
    if(deep[x] > deep[y])
        swap(x,y);
    while(deep[y] != deep[x])
    {
        ans += dis[y];
        y = fa[y];
    }
    while(x != y)
    {
        ans += dis[x];
        ans += dis[y];
        x = fa[x];
        y = fa[y];
    }
    return ans;
}
int main()
{
    scanf("%d",&n);
    memset(first,-1,sizeof(first));
    memset(dis,0x3f3f3f3f,sizeof(dis));
    for(int i = 1; i < n; i ++)
    {
        fa[i] = i;
        int f, t, d;
        scanf("%d%d%d", &f, &t, &d);
        build(f,t,d);
        build(t,f,d);
    }
    deep[0] = 1;
    vis[0] = 1;
    dfs(0); 
    scanf("%d",&m);
    while(m --)
    {
        scanf("%d%d", &f, &t);
        printf("%d\n",lca(f,t));
    }
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值