「模板」点的距离(LCA)

「模板」点的距离(LCA)

思路

模板题,乱搞即可

(本文选用方法：倍增求lca)

具体看代码注释

Code

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+10;
const int logN=40;
struct node
{
	int to;
	int nex;
}a[N];
int tot,head[N];
int vis[N];//标记数组,看f[i][j]有没有被计算 
int f[N][logN];//f[i][j]:节点i往上走2的j次方步能到达的节点 
int dep[N];//dep[i]:节点i的深度 
int n;
void add(int u,int v)
{
	a[++tot].nex=head[u];
	a[tot].to=v;
	head[u]=tot;
}
void pre(int u,int fa)//预处理f数组 
{
	dep[u]=dep[fa]+1;
	f[u][0]=fa;//走2的0次方(就是1)步到达他的父节点 
	vis[u]=1;//标记 
	for(int i=0;i<20;i++) f[u][i+1]=f[f[u][i]][i];
	for(int i=head[u];i;i=a[i].nex)
	{
		int v=a[i].to;
		if(vis[v]) continue;
		pre(v,u);
	}
}
int lca(int x,int y)//要求的节点x和节点y的最短距离 
{
	//默认x的深度更大 
	if(dep[x]<dep[y]) swap(x,y);
	for(int i=20;i>=0;i--)//先跳到同一深度 
	{
		if(dep[f[x][i]]>=dep[y]) x=f[x][i];
		if(x==y) return x;//一样了x就是最近公共祖先 
	}
	for(int i=20;i>=0;i--)//同时向上跳，跳到lca的下一层深度 
	{
		if(f[x][i]!=f[y][i])
		{
			x=f[x][i];
			y=f[y][i];
		}
	}
	return f[x][0];
}
int x,y;
int q;
int main()
{
	ios::sync_with_stdio(0);
	cin.tie(0);
//	memset(head,-1,sizeof(head));
	cin>>n;
	for(int i=1;i<n;i++)
	{
		cin>>x>>y;
		add(x,y);
		add(y,x);
	}
	pre(1,0);
	cin>>q;
	while(q--)
	{
		cin>>x>>y;
		cout<<(dep[x]+dep[y])-2*dep[lca(x,y)]<<endl;
	}
	return 0;
}