题目传送门https://vjudge.net/problem/CSES-1135#author=GPT_zh

解题思路

题目让我们求树上两个点的距离。

那么就可以转化为两点到其 LCA 的距离之和。

代码

#include<bits/stdc++.h>
using namespace std;

int n,q;
vector<int> g[200001];
int dis[200001],dep[200001],f[200001][31],lg[200001];
void dfs(int x,int fa)
{
	dep[x]=dep[fa]+1;
	f[x][0]=fa;
	for(int i=1;i<=lg[dep[x]];i++)
	{
		f[x][i]=f[f[x][i-1]][i-1];
	}
	for(auto y:g[x])
	{
		if(y!=fa)
		{
			dis[y]=dis[x]+1;
			dfs(y,x);
		}
	}
}
int solve(int x,int y,int a,int b)
{
	if(dep[x]<dep[y])swap(x,y);
	while(dep[x]>dep[y])
		x=f[x][lg[dep[x]-dep[y]]-1];
	if(x==y)
		return abs(dis[a]-dis[x])+abs(dis[b]-dis[x]);
	
	for(int k=lg[dep[x]]-1;k>=0;k--)
	{
		if(f[x][k]!=f[y][k])
		{
			x=f[x][k];
			y=f[y][k];
		}
	}
	
	return abs(dis[a]-dis[f[x][0]])+abs(dis[b]-dis[f[x][0]]);
}
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	cin>>n>>q;
	int u,v;
	for(int i=1;i<n;i++)
	{
		cin>>u>>v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	lg[0]=1;
	for(int i=1;i<=n;i++)
	{
		lg[i]=lg[i-1]+(i==(1<<lg[i-1]));
	}
	dis[1]=0;
	dfs(1,1);
	while(q--)
	{
		cin>>u>>v;
		cout<<solve(u,v,u,v)<<"\n";
	}
	return 0;
}