【景区导游——LCA】

题目

代码

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 1e5 + 10;
const int M = 2 * N;
int p[N][18], d[N], a[N];
ll dis[N][18]; //注意这里要开long long
int h[N], e[M], ne[M], idx, w[M];
int n, k;
void add(int a, int b, int c)
{
    e[idx] = b, ne[idx] = h[a], h[a] = idx, w[idx++] = c;
}
void dfs(int u)
{
    for (int i = h[u]; ~i; i = ne[i])
    {
        int j = e[i];
        if (d[j])
            continue;
        d[j] = d[u] + 1;
        p[j][0] = u;
        dis[j][0] = w[i];
        for (int k = 1; k <= 17; k++)
        {
            p[j][k] = p[p[j][k - 1]][k - 1];
            dis[j][k] = dis[j][k - 1] + dis[p[j][k - 1]][k - 1];
        }
        dfs(j);
    }
}
ll lca(int a, int b)
{
    ll retv = 0;

    if (d[a] < d[b])
        swap(a, b);
    for (int i = 17; i >= 0; i--)
    {
        if (d[p[a][i]] >= d[b])
        {
            retv += dis[a][i];
            a = p[a][i];
        }
    }

    if (a == b)
        return retv;

    for (int i = 17; i >= 0; i--)
    {
        if (p[a][i] != p[b][i])
        {
            retv += dis[a][i];
            retv += dis[b][i];
            a = p[a][i];
            b = p[b][i];
        }
    }

    retv += dis[a][0];
    retv += dis[b][0];

    return retv;
}
int main()
{
    memset(h, -1, sizeof h);

    cin >> n >> k;
    for (int i = 1; i < n; i++)
    {
        int a, b, c;
        cin >> a >> b >> c;
        add(a, b, c);
        add(b, a, c);
    }
    
    d[1] = 1;
    dfs(1);
    
    ll tmp = 0;
    cin >> a[1];
    for (int i = 2; i <= k; i++)
    {
        cin >> a[i];
        tmp += lca(a[i - 1], a[i]);
    }

    for (int i = 1; i <= k; i++)
    {
        ll ans = tmp;
        if (i == 1)
            ans -= lca(a[1], a[1 + 1]);
        else if (i == k)
            ans -= lca(a[k - 1], a[k]);
        else
        {
            ans -= lca(a[i - 1], a[i]);
            ans -= lca(a[i], a[i + 1]);
            ans += lca(a[i - 1], a[i + 1]);
        }

        cout << ans << ' ';
    }

    return 0;
}
相关推荐
霜绛1 小时前
机器学习笔记(四)——聚类算法KNN、Kmeans、Dbscan
笔记·算法·机器学习·kmeans·聚类
晨非辰2 小时前
#C语言——学习攻略:深挖指针路线(三)--数组与指针的结合、冒泡排序
c语言·开发语言·数据结构·学习·算法·排序算法·visual studio
zzywxc7872 小时前
编程算法在金融、医疗、教育、制造业等领域的落地案例
人工智能·算法·金融·自动化·copilot·ai编程
zzywxc7872 小时前
编程算法在金融、医疗、教育、制造业的落地应用。
人工智能·深度学习·算法·机器学习·金融·架构·开源
conkl3 小时前
构建 P2P 网络与分布式下载系统:从底层原理到安装和功能实现
linux·运维·网络·分布式·网络协议·算法·p2p
Shan12053 小时前
递归算法的一些具体应用
算法
paopaokaka_luck5 小时前
婚纱摄影管理系统(发送邮箱、腾讯地图API、物流API、webSocket实时聊天、协同过滤算法、Echarts图形化分析)
vue.js·spring boot·后端·websocket·算法·echarts
愚戏师6 小时前
机器学习(重学版)基础篇(算法与模型一)
人工智能·算法·机器学习
OEC小胖胖8 小时前
渲染篇(二):解密Diff算法:如何用“最少的操作”更新UI
前端·算法·ui·状态模式·web
找不到、了8 小时前
Java排序算法之<归并排序>
算法·排序算法