点的距离

最新推荐文章于 2022-01-08 20:53:52 发布

￡xilm€

最新推荐文章于 2022-01-08 20:53:52 发布

阅读量131

点赞数

文章标签：算法 c++

本文链接：https://blog.csdn.net/qq_52898168/article/details/122352303

版权

这篇博客主要介绍了如何处理树的最短路径查询问题，利用LCA（Lowest Common Ancestor）算法高效求解。文章给出了一个包含N个节点的树结构，通过建立边并进行深度优先搜索，实现树的构建。接着，通过动态规划方法计算每个节点的祖先，并提供了求解两点间路径长度的函数。最后，展示了样例输入和输出，帮助理解算法的使用。

摘要由CSDN通过智能技术生成

题目描述

给定一棵 nn 个点的树，QQ 个询问，每次询问点 xx 到点 yy 两点之间的距离。

输入格式

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

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

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

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

输出格式

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

样例

Input	Output
6 1 2 1 3 2 4 2 5 3 6 2 2 6 5 6	3 4

LCA详解：https://blog.csdn.net/jeryjeryjery/article/details/52853017

#include<bits/stdc++.h>
using namespace std;
const int N=2e5+6;
int cnt;
int first[N],nxt[N],to[N];
int f[N][21],deep[N];
void add(int u,int v)
{
	nxt[++cnt]=first[u];
	first[u]=cnt;
	to[cnt]=v;
}
void deal_first(int u,int father)
{
	deep[u]=deep[father]+1;
	for(int i=0; i<=19; i++)
	{
		f[u][i+1]=f[f[u][i]][i];
	}
	for(int e=first[u]; e; e=nxt[e])
	{
		int v=to[e];
		if(v==father)
			continue;
		f[v][0]=u;
		deal_first(v,u);
	}
}
int lca(int x,int y)
{
	if(deep[x]<deep[y]) swap(x,y);
	for(int i=20; i>=0; i--)
	{
		if(deep[f[x][i]]>=deep[y])
			x=f[x][i];
		if(x==y)
			return x;
	}
	for(int 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 dis(int x,int y)
{
	return deep[x]+deep[y]-2*deep[lca(x,y)];
}
int main()
{
	int i,j,k;
	int n,m,x,y,t;
	cin>>n;
	for(i=1; i<n; i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y),add(y,x);
	}
	deal_first(1,0);
	cin>>t;
	while(t--)
	{
		scanf("%d%d",&x,&y);
		printf("%d\n",dis(x,y));
	}
	return 0;
}