爆刷PAT（甲级）——之【1151】 LCA in a Binary Tree (30 分)——先序中序建树+LCA问题

此题也是去年考的时候不会。时隔一年才来补一下。

这题网上其实很多的讲解了。大概分成三种讲法流派，一种是以柳神为代表的非建树递归找父子关系派；一类是最通俗常规的建树+遍历递归找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;
}