详解洛谷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;
}

结语

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

相关推荐
weixin_446122463 分钟前
LinkedList剖析
算法
百年孤独_1 小时前
LeetCode 算法题解:链表与二叉树相关问题 打打卡
算法·leetcode·链表
我爱C编程2 小时前
基于拓扑结构检测的LDPC稀疏校验矩阵高阶环检测算法matlab仿真
算法·matlab·矩阵·ldpc·环检测
算法_小学生2 小时前
LeetCode 75. 颜色分类(荷兰国旗问题)
算法·leetcode·职场和发展
运器1232 小时前
【一起来学AI大模型】算法核心:数组/哈希表/树/排序/动态规划(LeetCode精练)
开发语言·人工智能·python·算法·ai·散列表·ai编程
算法_小学生2 小时前
LeetCode 287. 寻找重复数(不修改数组 + O(1) 空间)
数据结构·算法·leetcode
岁忧2 小时前
(LeetCode 每日一题) 1865. 找出和为指定值的下标对 (哈希表)
java·c++·算法·leetcode·go·散列表
alphaTao2 小时前
LeetCode 每日一题 2025/6/30-2025/7/6
算法·leetcode·职场和发展
ゞ 正在缓冲99%…2 小时前
leetcode67.二进制求和
算法·leetcode·位运算
YuTaoShao2 小时前
【LeetCode 热题 100】240. 搜索二维矩阵 II——排除法
java·算法·leetcode