此题也是去年考的时候不会。时隔一年才来补一下。
这题网上其实很多的讲解了。大概分成三种讲法流派,一种是以柳神为代表的非建树递归找父子关系派;一类是最通俗常规的建树+遍历递归找LCA方法派;还有一类是只贴代码没有讲解或是使用了RMQ倍增高级方法解决LCA问题等。
记此博客来简单的缕一缕自己的LCA思绪。
脑袋已经锈了很久了。
1、先谈一下,最常规的建树+递归遍历LCA方法,本题复杂度就是O(2MN+MN):
题意;给一个二叉树(并不是BST)的先序遍历和中序遍历,节点值不重复。然后给M个值对的查询,求它们的最近公共祖先LCA节点。
难点:一个数输出的时候要分成好几个情况,写起来比较蛋疼。
还有一个,先不谈是不是必要进行建树,我们姑且就要建树的话,如何写出先序遍历/后序遍历 + 中序遍历进行建树。这个不去看一下以及多写几遍,是会导致大悲剧的。很重要!
第三个是,在判断完毕待查校训数据是否存在有效之后,如何进行常规的LCA递归遍历二叉树去寻找最近共同祖先呢?思路是——LCA的共同祖先分成三种情况:A和B不在同一颗子树,那么A和B的共同“根爷爷”就是答案;A在B的下侧,那么B就是A的祖先(在一侧);反之亦然,B在A的下侧(同侧),那么A就是B的祖先。——如此以来,我们从根节点开始递归左子树以及右子树分别去找A或者B,如果先找到了A,那不就说明B在A的下面几层(所以才A找到的时候B还没找到);如果先找到了B而A没找到,那A就是祖先;如果在左右子树分别找到A和B,那就说明当前节点就是它们的“共同根祖先”。
详细看代码吧
Code:
#include<bits/stdc++.h>
#include<unordered_map>
using namespace std;
#define inf 10009
#define INF 0x3f3f3f3f
int n,m;
int preorder[inf],inorder[inf];
struct Node
{
Node *left,*right;
int value;
Node(int v)
{
value=v;
left=right=NULL;
}
};
int flag;//结果类型
Node* Preorder_Inorder_Create(int l1,int r1,int l2,int r2)//先序中序遍历建树
{
if(l2>r2)return NULL;
Node *root=new Node(preorder[l1]);
int i;
for(i=l2;inorder[i]!=preorder[l1];i++);
int len=i-l2;
root->left=Preorder_Inorder_Create(l1+1,l1+len,l2,i-1);
root->right=Preorder_Inorder_Create(l1+1+len,r1,i+1,r2);
return root;
}
Node* findFather(Node *root,int a,int b)//递归找公共祖先
{
if(!root)return NULL;
else if(root->value==a||root->value==b)return root;
Node *left=findFather(root->left,a,b);
Node *right=findFather(root->right,a,b);
if(left&&right){return root;}//左右子树
else if(!left)return right;//在一边——右边
else return left;//在一边——左边
}
int main()
{
int i,j,k;
cin>>m>>n;
for(i=0;i<n;i++)cin>>inorder[i];
for(i=0;i<n;i++)cin>>preorder[i];
Node *root=Preorder_Inorder_Create(0,n-1,0,n-1);
while(m--)
{
int a,b;
cin>>a>>b;
//判断数是否存在
for(i=0;i<n;i++)if(preorder[i]==a)break;
for(j=0;j<n;j++)if(preorder[j]==b)break;
bool exsit1=(i==n)?1:0;
bool exsit2=(j==n)?1:0;
if(exsit1&&exsit2){printf("ERROR: %d and %d are not found.\n",a,b);continue;}//都不在
else if(exsit1||exsit2){printf("ERROR: %d is not found.\n",exsit1?a:b);continue;}//某一个不在
Node *ans=findFather(root,a,b);
int value=ans->value;
if(value==a)printf("%d is an ancestor of %d.\n",a,b);
else if(value==b)printf("%d is an ancestor of %d.\n",b,a);
else printf("LCA of %d and %d is %d.\n",a,b,value);
}
return 0;
}