PAT A 1143 AC代码（BST树LCA问题算法详解）

这题可以结合1151来学习理解。

首先要知道BST树中序序列即数据点增序序列。

（所以原本基于中序顺序的比较就可以转为点的大小比较）

BST树LCA解法思路理解（画个树就很好理解了)：

确保U、V均存在于树上后

从根结点开始

1、若当前结点index大小在U，V之间，则当前结点index就是U、V的LCA

2、若U、V均小于index，index=当前结点的左子，重复1-3步

3、若U、V均大于index，index=当前结点的右子，重复1-3步

---

所以我按这个思路，配合先序和中序递归来做（测试点四超时了）

原始代码：

#include<iostream>
#include<vector>
#include<set>
#include<algorithm>
using namespace std;
vector<int> pre,in;
set<int> s;
int M,N;
int LCA(int root,int start,int end,int U,int V){
	if(start>end) return -1;
	int index=pre[root];
	int i=start;
	while(index!=in[i]) i++;
	if(index>=min(U,V)&&index<=max(U,V)){
		return index;
	}else if(index>max(U,V)){
		return LCA(root+1,start,i-1,U,V);
	}else{
		return LCA(root+i-start+1,i+1,end,U,V);
	}
	
}
int main(){
	cin>>M>>N;
	pre.resize(N);
	in.resize(N);
	for(int i=0;i<N;i++){
		scanf("%d",&pre[i]);
		s.insert(pre[i]);
	}
	in=pre;
	sort(in.begin(),in.end());
	for(int i=0;i<M;i++){
		int U,V;
		scanf("%d %d",&U,&V);
		if(s.find(U)==s.end()&&s.find(V)==s.end()){
			printf("ERROR: %d and %d are not found.\n",U,V);
			continue;
		}else if(s.find(U)==s.end()){
			printf("ERROR: %d is not found.\n",U);
			continue;
		}else if(s.find(V)==s.end()){
			printf("ERROR: %d is not found.\n",V);
			continue;
		}
		int lca=LCA(0，0，N-1,U,V);
		if(lca==U){
			printf("%d is an ancestor of %d.\n",U,V);
			continue;
		}else if(lca==V){
			printf("%d is an ancestor of %d.\n",V,U);
			continue;
		}else{
			printf("LCA of %d and %d is %d.\n",U,V,lca);
			continue;
		}
	}
	return 0;
}

---

 然后学习了柳神的思路，因为先序有 根-左-右 的顺序，配合上述BST树LCA算法的理解，可以得知

对于每一个点，将先序序列分成三部分来看，根-左树部分-右树部分，根是最先判断的（算法理解中的第1步），然后要么在左树部分（算法理解中的第2步），要么在右树部分（算法理解中的第3步）。

对于左树部分与右树部分，也可以同理分解，皆保证根最先，依此类推，我们可以得知：

！！！只要按照先序遍历的顺序进行判断，就可以保证第一个满足的就是LCA！！！

所以新的AC代码如下（大大减少了树很大的情况下的递归调用耗时）：

区别仅仅是LCA部分

#include<iostream>
#include<vector>
#include<set>
#include<algorithm>
using namespace std;
vector<int> pre,in;
set<int> s;
int M,N;
int LCA(int U,int V){
	for(auto it:pre){
		if(it<=max(U,V)&&it>=min(U,V)) return it;
	}
	return -1;
}
int main(){
	cin>>M>>N;
	pre.resize(N);
	in.resize(N);
	for(int i=0;i<N;i++){
		scanf("%d",&pre[i]);
		s.insert(pre[i]);
	}
	in=pre;
	sort(in.begin(),in.end());
	for(int i=0;i<M;i++){
		int U,V;
		scanf("%d %d",&U,&V);
		if(s.find(U)==s.end()&&s.find(V)==s.end()){
			printf("ERROR: %d and %d are not found.\n",U,V);
			continue;
		}else if(s.find(U)==s.end()){
			printf("ERROR: %d is not found.\n",U);
			continue;
		}else if(s.find(V)==s.end()){
			printf("ERROR: %d is not found.\n",V);
			continue;
		}
		int lca=LCA(U,V);
		if(lca==U){
			printf("%d is an ancestor of %d.\n",U,V);
			continue;
		}else if(lca==V){
			printf("%d is an ancestor of %d.\n",V,U);
			continue;
		}else{
			printf("LCA of %d and %d is %d.\n",U,V,lca);
			continue;
		}
	}
	return 0;
}