PAT 1151 LCA in a Binary Tree C++版

PAT 1151 LCA in a Binary Tree C++版

1.题意

给出一棵二叉树的中序遍历和前序遍历序列，让你求出这棵树中某两个子节点的最小祖宗节点（LCA）。

2.分析

这道题的主要思路有如下几种：
思路一：找出某个节点的所有父节点（按照顺序从上到下），然后使用两层for循环找到第一次相等的节点就是我们欲求的最小LCA。但是很遗憾，对于这道用时1000ms以内的都显得力不从心。
思路二：我们使用一次深搜，找出所有节点的parent节点并放入parent[]，并找出每个节点的当前深度并放入到depth[]中。对于两个节点u和v，我们找到parent[u]==parent[v]时，就说明已经找到公共最小祖宗了。
具体代码如下：

3.代码

#include<cstdio>
#include<iostream>
#include<vector>
#include<set>
#define maxn 0x3fffffff
using namespace std;

int in[maxn];
int pre[maxn];
set<int> tree;//保存二叉树的值 
int N,M;
int parent[maxn] ;//父亲节点(根节点的父亲记为-1) 
int depth[maxn] ;//节点的深度 

struct node{
	int data;
	node* lchild;
	node* rchild; 
};

node* create(int preL,int preR,int inL,int inR){
	if(preL > preR){
		return NULL;
	}
	node* root = new node;
	root->data = pre[preL];//将pre的preL值放入到结构体中
	tree.insert(root->data);//放入到tree中 
	int i ;
	for(i = inL; i <= inR;i++) {
		if(in[i] == pre[preL]) break;
	}
	int numLeft = i - inL;//看左边是否还有节点 
	//注意这里没有对numLeft的数目进行判断！！ 
	root->lchild = create(preL+1,preL+numLeft,inL,i-1);
	root->rchild = create(preL+numLeft+1,preR,i+1,inR);	
	return root;
}

//深搜设置 parent[]和 depth[] 
void dfs(node* root,int pare,int dep) {
	parent[root->data] = pare;//为当前节点设置父节点
	depth[root->data] = dep; //为当前节点设置深度 
	
	if(root->lchild != NULL) {//如果左子树不为空 
		dfs(root->lchild,root->data,dep +1);
	}		
	if(root->rchild != NULL) {
		dfs(root->rchild,root->data,dep +1);		
	}
}

bool find(int value){
	if(tree.find(value)!=tree.end()) return true;
	return false;
}

int lca(int u,int v){
	while(depth[u] > depth[v]) u = parent[u];//往上一层
	while(depth[v] > depth[u]) v = parent[v];//往上一层
	while(u!=v){//一直往上 
		u = parent[u];
		v = parent[v];
	}
	return u;	 
}

int main(){
	cin >> M >> N;
	int i,j;
	for(i = 0;i< N;i++) cin >> in[i];
	for(i = 0;i< N;i++) cin >> pre[i];
	node* root;
	root = create(0,N-1,0,N-1);	
	dfs(root,-1,0);
	
	int root_val = root->data;//表示根节点的值 	
	
	int que1,que2;//待查询的两个值 
	while(M){
		cin >> que1 >> que2;		
		int flag = 0;		
		if( find(que1) && find(que2) ) {//说明找到了值 ，进行查询 
			int res = lca(que1,que2);//返回的根节点的值 
			if(res == que1 ){
				cout <<res<<" is an ancestor of "<<que2<<".\n";
			}
			else if( res == que2){
				cout <<res<<" is an ancestor of "<<que1<<".\n";
			} 
			else 
			printf("LCA of %d and %d is %d.\n",que1,que2,res);
		}
		else{
			if(!find(que1) && !find(que2)){
				cout << "ERROR: "<< que1<<" and "<<que2<<" are not found.\n";
			}
			else if(!find(que1) && find(que2)){
				cout << "ERROR: "<< que1<<" is not found.\n";
			}
			else if(find(que1) && !find(que2)){
				cout << "ERROR: "<< que2<<" is not found.\n";
			}
		} 
		M--;
	}
}

4.测试用例

6 8
7 2 3 4 6 5 1 8
5 3 7 2 6 4 8 1
2 6
8 1
7 9
12 -3
0 8
99 99


0 8
7 2 3 4 6 5 1 8
5 3 7 2 6 4 8 1

1 8
7 2 3 4 6 5 1 8
5 3 7 2 6 4 8 1
2 6

5.执行结果

6.注意点

需要注意的点有：

• 因为给出的树的值的范围是int型，所以如果不开数组稍大的话，会造成段错误。于是我给数组开了个0x3fffffff。如果真的有2^31-1次方这样的数，如果还会造成段错误的话，那么直接用map吧。

7.参考知识

7.1求lca的具体方法