浙大版《数据结构学习与实验指导(第2版)》进阶实验8-2.3:二叉搜索树的最近公共祖先

浙大版《数据结构学习与实验指导(第2版)》进阶实验8-2.3:二叉搜索树的最近公共祖先,第1张

题意

给定一棵二叉搜索树的先序遍历序列,要求你找出任意两结点的最近公共祖先结点(简称 LCA)。

Input
输入的第一行给出两个正整数:待查询的结点对数 M M M ≤ 1000 \le 1 000 1000)和二叉搜索树中结点个数 N N N ≤ 10000 \le 10 000 10000)。

随后一行给出 N N N 个不同的整数,为二叉搜索树的先序遍历序列。

最后 M M M 行,每行给出一对整数键值 U U U V V V

所有键值都在整型int范围内。

Output
对每一对给定的 U U U V V V,如果找到 A 是它们的最近公共祖先结点的键值,则在一行中输出 LCA of U and V is A.。

但如果 U U U V V V 中的一个结点是另一个结点的祖先,则在一行中输出 X is an ancestor of Y.,其中 X 是那个祖先结点的键值,Y 是另一个键值。

如果 二叉搜索树中找不到以 U U U V V V 为键值的结点,则输出 ERROR: U is not found. 或者 ERROR: V is not found.,或者 ERROR: U and V are not found.。

思路

难点在于建树
二叉搜索树性质:左儿子的值<根节点的值<右儿子的值,所以中序遍历是递增的序列。


题目给出了先序遍历,那么从小到大排序后就是中序遍历,建树的过程位根据先序遍历和中序遍历建立二叉树。


递归建树后,根据二叉搜索树的性质递归查找lca的值。

如果x,y都大于当前节点的值,说明x,y都在当前节点的右子树;如果x,y都小于当前节点的值,说明x,y都在当前节点的左子树;否则,说明x,y分列在当前节点的左、右子树,那么当前节点就是lca。

代码
#include 
using namespace std;
const int maxn=1e4+100;
#define debug(x) cout<<#x<<":"<<x<<endl;

struct node{
    int val;
    node *l;
    node *r;
};
node tr[maxn];
int a[maxn],b[maxn],idx;
map<int,int>mp;

node* build(int len,int u,int st){
    if(len<1) return 0;
    node* now=&tr[idx++];///根节点
    now->val=a[u];
    int t=0;
    t=lower_bound(b+t+st,b+len+st,a[u])-b-st;///找到根节点
   /// debug(t);
    now->l=build(t,u+1,st);
    now->r=build(len-t-1,u+t+1,st+t+1);
    return now;
}

int lca(int x,int y,node* root){
    if(x>root->val&&y>root->val)
        return lca(x,y,root->r);
    else if(x<root->val&&y<root->val)
        return lca(x,y,root->l);
    else return root->val;
}

int main()
{
    int m,n;cin>>m>>n;
    for(int i=0;i<n;i++){
        cin>>a[i];
        ///debug(a[i]);
        b[i]=a[i];
        mp[b[i]]=1;///标记出现过
    }
    sort(b,b+n);///中序遍历
    node* root=build(n,0,0);
    while(m--){
        int x,y;
        cin>>x>>y;
        if(!mp.count(x)&&!mp.count(y)) printf("ERROR: %d and %d are not found.\n",x,y);
        else if(!mp.count(x)&&mp.count(y)) printf("ERROR: %d is not found.\n",x);
        else if(mp.count(x)&&!mp.count(y)) printf("ERROR: %d is not found.\n",y);
        else{
            int par=lca(x,y,root);
            if(par==x){
                printf("%d is an ancestor of %d.\n",x,y);
            }
            else if(par==y){
                printf("%d is an ancestor of %d.\n",y,x);
            }
            else{
                printf("LCA of %d and %d is %d.\n",x,y,par);
            }
        }
    }
    return 0;
}


欢迎分享,转载请注明来源:内存溢出

原文地址: http://outofmemory.cn/langs/673298.html

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2022-04-19
下一篇 2022-04-19

发表评论

登录后才能评论

评论列表(0条)

保存