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

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

题意

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


Input

输入的第一行给出两个正整数:待查询的结点对数 M(≤ 1000)和二叉搜索树中结点个数 N(≤ 10000)。随后一行给出 N  个不同的整数,为二叉搜索树的先序遍历序列。最后 M行,每行给出一对整数键值 U和 V。所有键值都在整型int范围内。


Output

对每一对给定的 U和 V,如果找到 A 是它们的最近公共祖先结点的键值,则在一行中输出 LCA of U and V is A.。但如果 U和 V中的一个结点是另一个结点的祖先,则在一行中输出 X is an ancestor of Y.,其中 X 是那个祖先结点的键值,Y 是另一个键值。如果 二叉搜索树中找不到以 U或 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 <bits/stdc++.h>
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;
}
目录
相关文章
|
18天前
|
存储 算法 数据安全/隐私保护
【Python学习篇】Python实验小练习——高级数据结构(五)
【Python学习篇】Python实验小练习——高级数据结构(五)
33 1
|
18天前
|
存储 算法 数据挖掘
数据结构实验||约瑟夫环
数据结构实验||约瑟夫环
|
11天前
|
存储 算法 程序员
【C++进阶】深入STL之 栈与队列:数据结构探索之旅
【C++进阶】深入STL之 栈与队列:数据结构探索之旅
18 4
|
19天前
|
存储 算法 C语言
数据结构学习记录——图-最短路径问题(无权图单源最短路径算法、有权图单源最短路径算法、多源最短路径算法、Dijkstra(迪杰斯特拉)算法、Floyd算法)
数据结构学习记录——图-最短路径问题(无权图单源最短路径算法、有权图单源最短路径算法、多源最短路径算法、Dijkstra(迪杰斯特拉)算法、Floyd算法)
23 1
|
19天前
|
存储 算法 测试技术
数据结构学习记录——树习题-Complete Binary Search Tree(题目描述、输入输出示例、数据结构的选择、核心算法、计算左子树的规模)
数据结构学习记录——树习题-Complete Binary Search Tree(题目描述、输入输出示例、数据结构的选择、核心算法、计算左子树的规模)
19 1
|
19天前
|
测试技术 C语言
数据结构学习记录——树习题—Tree Traversals Again(题目描述、输入输出示例、解题思路、解题方法C语言、解析)
数据结构学习记录——树习题—Tree Traversals Again(题目描述、输入输出示例、解题思路、解题方法C语言、解析)
15 1
|
19天前
|
存储
数据结构学习记录——如何建立图(邻接矩阵、邻接表-图节点的结构、创建并初始化、插入变、完整图的建立)
数据结构学习记录——如何建立图(邻接矩阵、邻接表-图节点的结构、创建并初始化、插入变、完整图的建立)
16 0
|
19天前
|
存储 算法
数据结构学习记录——图应用实例-六度空间(题目描述、算法思路、伪代码及解读、图解)
数据结构学习记录——图应用实例-六度空间(题目描述、算法思路、伪代码及解读、图解)
13 0
|
20小时前
|
存储 测试技术
【数据结构】操作受限的线性表,栈的具体实现
【数据结构】操作受限的线性表,栈的具体实现
11 5
|
1天前
|
算法
【C/数据结构和算法】:栈和队列
【C/数据结构和算法】:栈和队列
6 1