YbtOJ 倍增问题课堂过关例2 树上距离【倍增】【lca】

最新推荐文章于 2024-09-14 18:15:47 发布

JA_yichao

最新推荐文章于 2024-09-14 18:15:47 发布

阅读量113

点赞数

分类专栏： YbtOJ专项练习题题解 lca 文章标签： lca 倍增 YbtOJ 题解 c++

本文链接：https://blog.csdn.net/Jackma_mayichao/article/details/118567577

版权

题解同时被 3 个专栏收录

414 篇文章 2 订阅

订阅专栏

YbtOJ专项练习题

102 篇文章 2 订阅

订阅专栏

lca

4 篇文章 0 订阅

订阅专栏

这篇博客详细介绍了如何运用LCA（最近公共祖先）模板解决图论中的路径权值计算问题。代码中展示了如何构建树状数组并进行深度优先搜索来找到两点的最近公共祖先，并计算路径上的权值差。同时，博主强调了数组大小设置和避免数组越界的重要性。

摘要由CSDN通过智能技术生成

在这里插入图片描述

思路

lca模板题。
注意数组大小，稍微比范围大一点，不然怎么死的都不知道。

代码

#include<iostream>
#include<cstdio>
using namespace std;
int dep[100010],dis[100010],f[100010][25];
int hd[100010],tot;
int n,m,x,y,k;
struct node
{
	int y,next,z;
}a[100010];
void add(int x,int y,int w)
{
	a[++tot]=(node){y,hd[x],w};
	hd[x]=tot;
}
void finddep(int x,int fa)
{
	f[x][0]=fa;
	dep[x]=dep[fa]+1;
	for(int i=hd[x]; i; i=a[i].next)
	 {
	 	int yy=a[i].y;
	 	if(yy==fa)
	 	  continue;
	 	dis[yy]=dis[x]+a[i].z;
	 	finddep(yy,x);
	 }
}
int lca(int x,int y)
{
	if(dep[x]>dep[y])
	  swap(x,y);
	int p=dep[y]-dep[x],k=20,t=1<<k;
	while(p)
	 {
	 	if(t<=p)
	 	  p-=t,y=f[y][k];
	 	k--,t=1<<k;
	 }
	if(x==y)
	  return x;
    for(int i=20; i>=0; i--)
     {
     	if(f[x][i]!=f[y][i])
     	 {
     	 	x=f[x][i];
     	 	y=f[y][i];
		 }
	 }
	return f[x][0];
}
int main()
{
	cin>>n>>m;
	for(int i=1; i<=n-1; i++)
	 {
	 	scanf("%d%d%d",&x,&y,&k);
	 	add(x,y,k);
	 	add(y,x,k);
	 }
	finddep(1,0);
	for(int j=1; (1<<j)<=n; j++)
	 for(int i=1; i<=n; i++)
	    f[i][j]=f[f[i][j-1]][j-1];
	while(m--)
	 {
	 	scanf("%d%d",&x,&y);
	 	int ans=lca(x,y);
	 	printf("%d\n",dis[x]-dis[ans]+dis[y]-dis[ans]);
	 }
	return 0;
}