左神算法学习篇1：搜索二叉树相关

最新推荐文章于 2022-11-14 20:14:22 发布

香农派我最爱

最新推荐文章于 2022-11-14 20:14:22 发布

阅读量292

点赞数 2

分类专栏：左神算法学习笔记文章标签：二叉树数据结构算法 leetcode

本文链接：https://blog.csdn.net/weixin_43161461/article/details/111401361

版权

左神算法学习笔记专栏收录该内容

2 篇文章 0 订阅

订阅专栏

Day 1：搜索二叉树相关

在这里插入图片描述

题目

已知一个搜索二叉树后序遍历的数组posArr，请根据posArr，重建出整颗树返回新建树的头节点。

解题思路一

首先搜索二叉树BST定义，左子节点<根节点<右子节点，其次后序遍历为左右根，这些知识在之前相关文章中都已提到过。所以我们可以判断出这个posArr最右边的值为当前子树的根节点！（左右跟嘛，根节点一定在最右边），下面除去最右边的根节点后，数组剩余元素就是左右子树的节点了。那么有以下三种情况：

1.数组剩余部分包含左右子树，这时需要去寻找左右子树的分界点，依据BST概念来从左向右依次遍历查找，在区间[L,R-1]中找到中点M，那么左子树就是[L,M]，右子树[M+1,R-1]。

2.数组剩余部分为左子树，那么这时，遍历查找后M==R-1，当前只有左子树[L,R-1]，右子树为null。

3.数组剩余部分为右子树，遍历查找后M为初始值（因为右子树节点都是大于根节点的），这时当前只含右子树[L,R-1]，左子树为null。

注：方法二的代码将M初始值设为-1，后面依据查找过后M值得情况再做分别判断比较清晰。方法一则将M==L-1，这个比较巧妙，主要体现在区间只有左或右子树时，即只有左子树时，M为R-1，那么process1(posArr,M+1,R-1)正好出现左区间大于右区间返回null，当只有右子树时，M为初始值L-1，那么process1(posArr,L,M)出现左区间大于右区间情况返回null。
因为每次区间的中间分界点查找都是遍历，所以时间复杂度为O(n^2)。

代码实现

	public static Node posArrayToBST(int[] posArr){
        return process1(posArr,0,posArr.length-1);
    }

    public static Node process1(int[] posArr,int L,int R){//O(n^2)
        if(L>R){
            return null;
        }
        Node head=new Node(posArr[R]);
        if (L==R){
            return head;
        }
        int M=L-1;
        for(int i=L;i<R;i++){
            if (posArr[i]<posArr[R]){
                M=i;
            }
        }
        head.left=process1(posArr,L,M);
        head.right=process1(posArr,M+1,R-1);
        return head;
    }


    public static Node process2(int[] posArr,int L,int R){//O(n^2)
        if(L>R){
            return null;
        }
        Node head=new Node(posArr[R]);
        if (L==R){
            return head;
        }
        int M=-1;
        for(int i=L;i<R;i++){
            if (posArr[i]<posArr[R]){
                M=i;
            }
        }
        if (M==-1){//>
            head.right=process2(posArr,L,R-1);
        }else if(M==R-1){
            head.left=process2(posArr,L,R-1);
        }else {
            head.left=process2(posArr,L,M);
            head.right=process2(posArr,M+1,R-1);
        }
        return head;
    }

解题思路二

下面讲解一直时间复杂度更低的解法，每次在查找左右子树的分界点时可以利用二分查找来找出分界节点。这样每次查找时间复杂度由原来的遍历O(n)变为二分的O(logn)，所以总的时间复杂度为O(nlogn)。

代码实现

	public static Node process3(int[] posArr,int L,int R){//O(nlogn)
        if(L>R){
            return null;
        }

        Node head=new Node(posArr[R]);
        if (L==R){
            return head;
        }
        int M=L-1;
        int left=L;
        int right=R-1;
        while (left<=right){
            int mid=left+(right-left)>>1;
            if(posArr[mid]<posArr[R]){
                M=mid;
                left=mid+1;
            }else {
                right=mid-1;
            }
        }
        head.left=process3(posArr,L,M);
        head.right=process3(posArr,M+1,R-1);
        return head;
    }