1151 LCA in a Binary Tree (30 分)

最新推荐文章于 2024-10-02 14:37:57 发布

念卿_qing

最新推荐文章于 2024-10-02 14:37:57 发布

阅读量78

点赞数

分类专栏： PAT 文章标签： c语言

本文链接：https://blog.csdn.net/kk_qing/article/details/120021238

版权

PAT 专栏收录该内容

26 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何通过中序遍历和先序遍历来找到二叉树中两个给定节点的最近公共祖先。文章首先分析了问题的解决策略，即判断节点在根节点的左右两侧，并利用映射存储节点位置。接着，给出了C++代码实现，通过递归地查找公共祖先。最后，代码处理了不同边界条件，确保了正确性。

摘要由CSDN通过智能技术生成

题目

题意

给出中序和先序序列，判断给定两个结点的最近公共祖先

分析

仅分析两结点都在树上的情况。
对一颗树上的两个结点而言，要么两个结点都在根结点的左子树或右子树，要么一左一右。如果一左一右，则两结点的公共祖先即为当前根结点；若都在左/右子树，则在左/右子树查询结点和根结点的相对位置。

关键点

使用map建立结点key值和位置间的映射，为判断待查询结点a、b与根结点的相对位置提供方便。

代码

#include <bits/stdc++.h>
using namespace std;
vector<int> pre(10000),in(10000);
map<int,int> mp;
void lca(int prel,int inl,int inr,int a,int b){
    if(inl>inr) return;
    int inroot=mp[pre[prel]],ina=mp[a],inb=mp[b];
    if(ina<inroot&&inb<inroot) lca(prel+1,inl,inroot-1,a,b);
    else if((ina>inroot&&inb<inroot)||(ina<inroot&&inb>inroot)){
        printf("LCA of %d and %d is %d.\n",a,b,in[inroot]);
    }
    else if(ina>inroot&&inb>inroot){
        lca(prel+1+inroot-inl,inroot+1,inr,a,b);
    }
    else if(ina==inroot) printf("%d is an ancestor of %d.\n",a,b);
    else if(inb==inroot) printf("%d is an ancestor of %d.\n",b,a);
}
int main()
{
    int m,n;
    scanf("%d %d",&m,&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&in[i]);
        mp[in[i]]=i;
    }
    for(int i=1;i<=n;i++) scanf("%d",&pre[i]);
    while(m--){
        int a,b;
        scanf("%d %d",&a,&b);
        if(mp[a]==0&&mp[b]==0) printf("ERROR: %d and %d are not found.\n",a,b);
        else if(mp[a]==0||mp[b]==0) printf("ERROR: %d is not found.\n",mp[a]==0?a:b);
        else if(mp[a]!=0&&mp[b]!=0){
            lca(1,1,n,a,b);
        }
    }
}