【Ybtoj 第18章例3】树上距离【倍增问题】

---

解题思路

我们需要求出树上任意两点间的距离，那么我们可以任意前一个节点作为根节点，求出根节点到所有点的距离，用dis数组表示，用$len(a,b)$表示点对$（a,b）$之间的距离，
则$len(a,b)=di{s}_x+di{s}_y-2\ast di{s}_{L}CA(a,b)$

》》LCA的模板题

---

代码

#include<iostream>
#include<cstdio>
#include<iomanip>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;

int n,m,x,y,k,kk,lca,dep[50010],lg[50010],head[100010],f[50010][25],dis[50010];

struct c{
	int x,next,w;
}a[100010];

void add(int x,int y,int k)
{
	a[++kk].x=y;
	a[kk].next=head[x];
	a[kk].w=k;
	head[x]=kk;
}

void dfs(int x,int fa){
	f[x][0]=fa;
	for(int i=1;i<=lg[dep[x]];i++)
		f[x][i]=f[f[x][i-1]][i-1];
	for(int i=head[x];i;i=a[i].next){
		int y=a[i].x;
		if(y==fa)continue;
		dep[y]=dep[x]+1;
		dis[y]=dis[x]+a[i].w;
		dfs(y,x);
	}
	return ;
}

int LCA(int x,int y){
	if(dep[x]>dep[y])swap(x,y);
	while(dep[y]>dep[x])
		y=f[y][lg[dep[y]-dep[x]]];
	if(x==y)return x;
	for(int i=lg[dep[x]];i>=0;i--)
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[y][0];
}

int main(){
	scanf("%d%d",&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);
	}
	dep[1]=1;
	lg[0]=-1;
	for(int i=1;i<=n;i++)
		lg[i]=lg[i>>1]+1;
	dfs(1,0);
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		lca=LCA(x,y);
		printf("%d\n",dis[x]+dis[y]-dis[lca]*2);
	}
}