二叉查找树（重复元素情况）

仰望星空的鑫

于 2022-06-02 22:17:10 发布

阅读量1.9k

点赞数

分类专栏：力扣刷题文章标签：排序算法 leetcode 算法

本文链接：https://blog.csdn.net/weixin_53700782/article/details/125107457

版权

力扣刷题专栏收录该内容

2 篇文章 0 订阅

订阅专栏

二叉树查找元素的位置（含有重复元素，且元素个数是不确定的）

代码解析

package 在排序数组中查找第一个元素与最后一个元素;

public class Solution {
    public int[] searchRange(int[] nums, int target) {
        int[] result =new int[2];
        result[0] = findFirstLeft2(nums,target);
        result[1] = findFirstRight2(nums,target);
        return result;
    }
    /*
    * 查找第一个与最后一个元素，说明排序数组是含有重复元素的（原来的二叉树是不含相同元素的）
    * 所以这里要做一些小的调整
    * */
//正常不包含重复元素时利用二分查找法查找元素
    public static int findFirstLeft(int[] nums, int target){
        int left = 0;
        int right = nums.length-1;
        while (left <= right){
            int mid = left+(right-left+1)/2;
            //比较目标值和中间值的大小；
            if(nums[mid] > target){
                //说明目标值在mid的左边，缩小范围
                right = mid-1;
            }else if(nums[mid] < target){
                left = mid+1;
            }else {
                return mid;
            }
        }
        return -1;
    }
//包含相同元素，则要判断该元素是首元素还是末元素。
    public static int findFirstLeft2(int[] nums , int target){
        int left = 0;
        int right = nums.length-1;
        while (left <= right){
            int mid = left+(right-left+1)/2;
            //比较目标值和中间值的大小；
            if(nums[mid] > target){
                //说明目标值在mid的左边，缩小范围
                right = mid-1;
            }else if(nums[mid] < target){
                left = mid+1;
            }else {
                //因为该数组是按照升序排序的，所以只要判断该元素
                //前一个元素是否相同，就知道该元素是第一个还是第二个
                if(mid == 0 || nums[mid] != nums[mid-1]){//mid == 0是防止mid-1超出数组范围。
                    //说明mid就是第一个元素
                    return mid;
                }else {
                    //说明该元素前面还有相同的元素，但是具体有几个，我们不秦楚
                    //所以我们不能直接返回前一个元素的位置，而是缩小范围，继续找第一个元素。
                    right = mid-1;
                }
            }
        }
        return -1;
    }

    //查找最后一个元素的位置
    public static int findFirstRight2(int[] nums , int target){
        int left = 0;
        int right = nums.length-1;
        while (left <= right){
            int mid = left+(right-left+1)/2;
            //比较目标值和中间值的大小；
            if(nums[mid] > target){
                //说明目标值在mid的左边，缩小范围
                right = mid-1;
            }else if(nums[mid] < target){
                left = mid+1;
            }else {
                //因为该数组是按照升序排序的，所以只要判断该元素
                //前一个元素是否相同，就知道该元素是第一个还是第二个
                if(mid == nums.length-1 || nums[mid] != nums[mid+1]){//如果mid为最后一个元素，就不存在下一个元素了
                    //说明mid就是第一个元素
                    return mid;
                }else {
                    //说明该元素前面还有相同的元素，但是具体有几个，我们不秦楚
                    //所以我们不能直接返回前一个元素的位置，而是缩小范围，继续找第一个元素。
                    left = mid+1;
                }
            }
        }
        return -1;
    }


}