POJ1330最近公共祖先LCA

题目大意:
给你一棵树,找出两点的最近公共祖先LCA。
输入:第1行包含一个整数T,表示测试用例的数量。每个测试用例的第1行都包含整数N(2≤N≤10,000),表示树中的节点数。节点用1~N标记。接下来N-1行,每行都包含一对边的整数,第1个整数是第2个整数的父节点(有N个节点的树有N-1条边)。每个测试用例的最后一行都包含两个不同的整数,求其LCA。
输入样例
2
16
1 14
8 5
10 16
5 9
4 6
8 4
4 10
1 13
6 15
10 11
6 7
10 2
16 3
8 1
16 12
16 7
5
2 3
3 4
3 1
1 5
3 5
输出样例:
4
3
题解:由于本题数据量不大,所以可以暴力求解最近公共祖先LCA。

#include<cstdio>
using namespace std;
const int maxn=10010;
int fa[maxn];//树的存储,使用父亲表示法,把一条边的信息存储在子节点上,fa[i]=j,表示子节点的父亲是i 
bool flag[maxn];

void Init(int n){
	for(int i=1;i<=n;i++){
		fa[i]=i;
		flag[i]=0;
	} 
}

int LCA(int u,int v){
    if(u==v)
    	return u;
	flag[u]=1;
	while(fa[u]!=u){//u向上走到根,并标记沿途进过的节点 
		u=fa[u];
		flag[u]=1;
	} 
	if(flag[v]) //如果v节点被访问过,那么v就是u,v的最近公共祖先 
		return v;
	while(fa[v]!=v){//v向上,找沿途被访问过的节点,访问过的节点就是u,v的最近公共祖先 
		v=fa[v];
		if(flag[v])
			return v;
	}
	return 0;   	
}

int main(){
	int n,u,v,T;
	scanf("%d",&T);
	while(T--){
		scanf("%d",&n);
		Init(n);
		for(int i=1;i<n;i++){
			scanf("%d%d",&u,&v);
			fa[v]=u;
		}
		scanf("%d%d",&u,&v);
		printf("%d\n",LCA(u,v));
	}
	return 0;
}
04-28
### POJ 1330 题解 POJ 1330 是一道经典的最近公共祖先 (Least Common Ancestor, LCA) 问题。该题的核心在于如何高效地求解两节点之间的最近公共祖先。 #### 倍增算法简介 倍增方法是一种高效的在线求解 LCA 的方式,其时间复杂度为 \(O((n+q)\log n)\),其中 \(n\) 表示树中的节点数量,\(q\) 表示查询次数。此方法通过预处理的方式记录每个节点向上跳跃若干步后的父节点位置,从而加速查询过程。 具体来说,定义数组 `fa[i][j]` 表示从节点 \(i\) 向上跳 \(2^j\) 步所到达的节点编号。为了支持这一操作,我们需要满足如下递推关系: \[ \text{fa}[i][j] = \begin{cases} \text{parent}(i), & j = 0 \\ \text{fa}[\text{fa}[i][j-1]][j-1], & j > 0 \end{cases} \] 这种预处理可以通过深度优先搜索 (DFS) 完成,在 DFS 过程中同时计算每个节点的深度以及它们的倍增父节点表。 #### 查询阶段 在实际查询过程中,假设我们要查找节点 \(a\) 和 \(b\)最近公共祖先,则分为以下几个部分: 1. **调整深度一致** 如果当前两个节点的深度不相同,则将较深的节点不断向上提升至两者深度相等的位置。这一步利用了倍增数组 `fa[i][j]` 来快速跳跃多个层次。 2. **同步爬升** 当两个节点处于同一深度时,让它们逐步向上移动直至相遇于某个共同祖先处。为了避免逐层遍历带来的效率低下问题,同样采用倍增策略按指数级跳跃。 最终返回的结果即为最后一次未分叉前的状态作为答案。 以下是基于以上逻辑的一个简单实现代码片段: ```cpp #include <bits/stdc++.h> using namespace std; const int MAXN = 5e4 + 5; vector<int> adj[MAXN]; int depth_[MAXN]; // 存储每个结点的深度 int fa[MAXN][20]; // 记录每个结点向上跳2^k步的父亲 void dfs(int u, int parent){ depth_[u]=depth_[parent]+1; fa[u][0]=parent; for(auto v : adj[u]){ if(v != parent){ dfs(v,u); } } } // 初始化fa[][]表格 void preprocess(int N){ memset(fa,-1,sizeof(fa)); for(int k=1;k<20;k++){ for(int i=1;i<=N;i++){ if(fa[i][k-1]!=-1){ fa[i][k]=fa[fa[i][k-1]][k-1]; } } } } int get_lca(int a,int b){ if(depth_[a]<depth_[b]) swap(a,b); // 将a提到和b相同的高度 for(int k=19;k>=0;k--){ if(fa[a][k]!=-1 && depth_[fa[a][k]] >= depth_[b]){ a=fa[a][k]; } } if(a==b)return a; // 同步爬升 for(int k=19;k>=0;k--){ if(fa[a][k]!=-1 && fa[b][k]!=-1 && fa[a][k]!=fa[b][k]){ a=fa[a][k]; b=fa[b][k]; } } return fa[a][0]; } ``` 上述程序实现了完整的倍增法用于解决LCA问题的功能模块化设计[^2]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值