LOJ #10130 点的距离

在LOJ做的第一道题。

最开始想复杂了qwq 想的是在求LCA的过程中统计向上的步数

其实此题很裸……就是求出u,v的LCA，

再分别用两点深度减去LCA的深度，再加起来就好了qwq
---化简---

\(dep[u]+dep[v]-2*dep[LCA(u,v)]\)

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#define MAXN 100005
using namespace std;
struct qwq
{
	int t,nex;
};
int tot=0,dep[MAXN],fa[MAXN][20],lg[MAXN],h[MAXN],n,m;
qwq e[MAXN<<1];
inline void add(int x,int y)
{
	e[++tot].t=y;
	e[tot].nex=h[x];
	h[x]=tot;
}
void dfs(int f,int fat)
{
	dep[f]=dep[fat]+1;
	fa[f][0]=fat;
	for (int i=1;(1<<i)<=dep[f];i++)
	{
		fa[f][i]=fa[fa[f][i-1]][i-1];
	}
	for (int i=h[f];i>0;i=e[i].nex)
	{
		if (e[i].t!=fat)
		{
			dfs(e[i].t,f);
		}
	}
}
int LCA(int x,int y)
{
	if (dep[x]<dep[y])
	{
		swap(x,y);
	}
	while (dep[x]>dep[y])
	{
		x=fa[x][lg[dep[x]-dep[y]]-1];
	}
	if (x==y)
	{
		return x;
	}
	for (int i=lg[dep[x]-1];i>=0;i--)
	{
		if (fa[x][i]!=fa[y][i])
		{
			x=fa[x][i];
			y=fa[y][i];
		}
	}
	return fa[x][0];
}
int maxn=-10;
inline void init()
{
	scanf("%d",&n);
	for (int i=1,x,y;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		maxn=max(maxn,max(x,y));
		add(x,y);
		add(y,x);
	}
	for (int i=1;i<=n;i++)
	{
		lg[i]=lg[i-1]+(1<<lg[i-1]==i);
	}
	dfs(maxn,0);
	scanf("%d",&m);
	for (int i=1,x,y;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		int lca=LCA(x,y);
		printf("%d\n",dep[x]+dep[y]-2*dep[lca]);
	}
}
int main()
{
	init();
	return 0;
}