LCA - 点的距离 - 一本通 4.4 例 1

题目描述

给定一棵 n 个点的树，Q 个询问，每次询问点 x 到点 y 两点之间的距离。

输入格式

第一行一个正整数 n，表示这棵树有 n 个节点；

接下来 n-1 行，每行两个整数 x,y 表示 x,y 之间有一条连边；

然后一个整数 Q，表示有 Q 个询问；

接下来 Q 行每行两个整数 x,y 表示询问 x 到 y 的距离。

输出格式

输出 Q 行，每行表示每个询问的答案。

样例

Input

6
1 2
1 3
2 4
2 5
3 6
2
2 6
5 6

Output

3
4

数据范围与提示

对于全部数据，1≤n≤10^5 ,1≤x,y≤n。

解题思路

倍增法，用一个函数来处理dep为节点的深度，f[a][b] 中存储a的2^b的父节点。LCA函数来查询两节点的最近公共父节点。两点距离则为ab深度减去2*dep[lca(a,b)]。

完整代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
vector<int>q[N];
int dep[N],f[N][21];
void deal(int a,int b)
{
	int i;
	dep[a]=dep[b]+1;
	for(i=0;i<=19;i++)
		f[a][i+1]=f[f[a][i]][i];
	for(i=0;i<q[a].size();i++)
	{
		int u=q[a][i];
		if(u==b) continue;
		f[u][0]=a;//u的父亲是a 
		deal(u,a);
	}
}
int lca(int x,int y)
{
	int i;
	if(dep[x]<dep[y]) swap(x,y);
	for(i=20;i>=0;i--)
	{
		if(dep[f[x][i]]>=dep[y]) x=f[x][i];
		if(x==y) return x;
	}
	for(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()
{
	int n,m,i,a,b;
	scanf("%d",&n);
	for(i=1;i<n;i++)
	{
		scanf("%d %d",&a,&b);
		q[a].push_back(b);
		q[b].push_back(a);
	}
	deal(1,0);
	scanf("%d",&m);
	while(m--)
	{
		scanf("%d %d",&a,&b);
		int sum=dep[a]+dep[b]-2*dep[lca(a,b)];
		printf("%d\n",sum);
	}
}