P4281 [AHOI2008]紧急集合 / 聚会

最新推荐文章于 2024-08-25 17:47:35 发布

crazy morning

最新推荐文章于 2024-08-25 17:47:35 发布

阅读量75

点赞数

分类专栏： LCA

本文链接：https://blog.csdn.net/weixin_49843717/article/details/115218049

版权

LCA 专栏收录该内容

9 篇文章 0 订阅

订阅专栏

这段代码实现了一个利用 Lowest Common Ancestor (LCA) 算法解决路径查询问题的程序。它通过DFS预处理建立从每个节点到根节点的路径，并在询问时快速找到两个节点的最近公共祖先。在询问三个节点x、y和z时，计算它们的LCA并输出路径长度之和的差值作为答案。

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

题目

思路

显然的LCA
code：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<cmath> 
using namespace std;
int fa[500010][27],head[500010],c[500010],x,y,z;
struct f{
	int to,net;
} a[1000020];
int n,m;
void dfs(int x,int f)
{
	fa[x][0]=f;
	c[x]=c[f]+1;
	for (int j=1;(1<<j)<=c[x];j++)
	{
		fa[x][j]=fa[fa[x][j-1]][j-1];
	}
	for (int i=head[x];i;i=a[i].net)
	{
		if (a[i].to!=f) dfs(a[i].to,x);
	}
	return;
}
int k=1;
void add(int u,int v)
{
    a[k].to=v;
	a[k].net=head[u];
    head[u]=k++;
    return; 
}
int LCA(int x,int y)
{
	if (c[x]>c[y]) swap(x,y);
	for (int i=20;i>=0;i--)
	{
		if (c[x]<=c[y]-(1<<i)) y=fa[y][i];
	}
	if (x==y) return x;
	for (int i=20;i>=0;i--)
	{
		if (fa[x][i]==fa[y][i]) continue;
		else x=fa[x][i],y=fa[y][i];
	}
	return fa[x][0];
}
long long ans;
int main()
{
	scanf("%d%d",&n,&m);
	for (int i=1;i<n;i++)
	{
		scanf("%d%d",&x,&y);
		add(x,y);
		add(y,x);
	}
	dfs(1,0);
	while (m--)
	{
		scanf("%d%d%d",&x,&y,&z);
		int xx=LCA(x,y),yy=LCA(y,z),zz=LCA(x,z);
		if (xx==yy) printf("%d ",zz);
		else if (yy==zz) printf("%d ",xx);
		else if (zz==xx) printf("%d ",yy);
		ans=c[x]+c[y]+c[z]-c[xx]-c[yy]-c[zz];
		printf("%lld\n",ans);
	}
    return 0;
}