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

结语

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

相关推荐
不見星空2 分钟前
leetcode 每日一题 1865. 找出和为指定值的下标对
算法·leetcode
我爱Jack13 分钟前
时间与空间复杂度详解:算法效率的度量衡
java·开发语言·算法
DoraBigHead2 小时前
小哆啦解题记——映射的背叛
算法
Heartoxx2 小时前
c语言-指针与一维数组
c语言·开发语言·算法
孤狼warrior2 小时前
灰色预测模型
人工智能·python·算法·数学建模
京东云开发者2 小时前
京东零售基于国产芯片的AI引擎技术
算法
chao_7893 小时前
回溯题解——子集【LeetCode】二进制枚举法
开发语言·数据结构·python·算法·leetcode
十盒半价4 小时前
从递归到动态规划:手把手教你玩转算法三剑客
javascript·算法·trae
GEEK零零七4 小时前
Leetcode 1070. 产品销售分析 III
sql·算法·leetcode
凌肖战4 小时前
力扣网编程274题:H指数之普通解法(中等)
算法·leetcode