详解洛谷P2912 [USACO08OCT] Pasture Walking G(牧场行走)(lca模板题)

题目

思路

一道模板题,没啥好说的,直接见代码

代码

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
int n,q,a,to[100001][22],b,deep[100001],c,t[1000001];
struct ff
{
  int id,len;
};
vector<ff> vec[100001];
void dfs(int x,int fa,int dp,int now)//now是x的深度(不计边权,根的深度计为1) dp是x距离根节点的距离(计边权)
{
  deep[x] = now;//单独开now变量是因为边权并不影响lca的值
  t[x] = dp;
  to[x][0] = fa;
  for(int i = 0;i < vec[x].size();i++)
    if(vec[x][i].id != fa)
      dfs(vec[x][i].id,x,dp + vec[x][i].len,now + 1);
}
int lca(int x,int y)
{
  if(deep[x] < deep[y]) swap(x,y);
  for(int i = 21;i >= 0;i--)
    if(deep[to[x][i]] >= deep[y])
      x = to[x][i];
  if(x == y) return x;
  for(int i = 21;i >= 0;i--)
    if(to[x][i] != to[y][i])
    {
      x = to[x][i];
      y = to[y][i];
    }
  return to[x][0];
}
int main()
{
  cin>>n>>q;
  for(int i = 1;i < n;i++)
  {
    scanf("%d%d%d",&a,&b,&c);
    vec[a].push_back({b,c});
    vec[b].push_back({a,c});
  }
  dfs(1,0,0,1);
  for(int i = 1;i <= 21;i++)
    for(int j = 1;j <= n;j++)
      to[j][i] = to[to[j][i - 1]][i - 1]; 
  while(q--)
  {
    scanf("%d%d",&a,&b);
    int k = lca(a,b);
    printf("%d\n",t[a] + t[b] - 2 * t[k]);
    //a的深度-lca(a,b)的深度就是a到lca(a,b)的距离,b的深度-lca(a,b)的深度就是b到lca(a,b)的距离
    //加起来就是a到b的距离(因为a->b的路径必定会经过lca(a,b))
  }
  return 0;
}

结语

如果这篇文章对您要帮助的话,请点赞支持一下吖! <(^-^)>

相关推荐
LNTON羚通1 小时前
摄像机视频分析软件下载LiteAIServer视频智能分析平台玩手机打电话检测算法技术的实现
算法·目标检测·音视频·监控·视频监控
哭泣的眼泪4082 小时前
解析粗糙度仪在工业制造及材料科学和建筑工程领域的重要性
python·算法·django·virtualenv·pygame
IT古董3 小时前
【机器学习】机器学习中用到的高等数学知识-8. 图论 (Graph Theory)
人工智能·机器学习·图论
Microsoft Word3 小时前
c++基础语法
开发语言·c++·算法
天才在此3 小时前
汽车加油行驶问题-动态规划算法(已在洛谷AC)
算法·动态规划
莫叫石榴姐4 小时前
数据科学与SQL:组距分组分析 | 区间分布问题
大数据·人工智能·sql·深度学习·算法·机器学习·数据挖掘
茶猫_5 小时前
力扣面试题 - 25 二进制数转字符串
c语言·算法·leetcode·职场和发展
肥猪猪爸7 小时前
使用卡尔曼滤波器估计pybullet中的机器人位置
数据结构·人工智能·python·算法·机器人·卡尔曼滤波·pybullet
readmancynn7 小时前
二分基本实现
数据结构·算法
萝卜兽编程7 小时前
优先级队列
c++·算法