代码随想录:二分查找相关题目推荐(35、34)

35.搜索插入位置

题目

给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。

请必须使用时间复杂度为 O(log n) 的算法。

示例 1:

复制代码
输入: nums = [1,3,5,6], target = 5
输出: 2

示例 2:

复制代码
输入: nums = [1,3,5,6], target = 2
输出: 1

示例 3:

复制代码
输入: nums = [1,3,5,6], target = 7
输出: 

代码(暴力遍历)

java 复制代码
class Solution {
    public int searchInsert(int[] nums, int target) {
        for(int i = 0; i < nums.length; i++){
            if(nums[i] >= target){
                return i;  //在数组最前面or直接找到or在某个数后面插入
            }
        }
        return nums.length;  //target最大,插入数组后面
    }
}

代码(二分查找标准版)

java 复制代码
class Solution {
    public int searchInsert(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1;
        int mid;
        while(left <= right){
            mid = left + (right - left)/2;
            if(target < nums[mid]){
                right = mid - 1;
            }
            else if(target > nums[mid]){
                left = mid + 1;
            }
            else{
                return mid;
            }
        }
        return right + 1;  //right指向待插入元素的前一个索引位置
    }
}

代码(二分查找自己写的)

java 复制代码
class Solution {
    public int searchInsert(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1;
        int mid;
        int index = -1;  //返回结果,必须要初始化,-1可以改成别的数

        while(left <= right){
            //特殊情况1:target在最前面插入
            if(target < nums[left]){
                index = left;  //插在最小值left位置
                break;
            }
            //特殊情况2:target在最后面插入
            if(target > nums[right]){
                index = right + 1;  //插在最大值right后面位置,不要忘+1
                break;
            }
            //二分查找逻辑
            mid = left + (right - left) / 2;
            if(target < nums[mid]){
                right = mid - 1;
            }
            else if(target > nums[mid]){
                left = mid + 1;
            }
            else{
                index = mid;  //找到了查找元素
                break;
            }
        }
        return index;
    }
}

34.在排序数组中查找元素的第一个位置和最后一个位置

题目

给你一个按照非递减顺序排列的整数数组 nums,和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target,返回 [-1, -1]

你必须设计并实现时间复杂度为 O(log n) 的算法解决此问题。

示例 1:

输入:nums = [5,7,7,8,8,10], target = 8
输出:[3,4]

示例 2:

输入:nums = [5,7,7,8,8,10], target = 6
输出:[-1,-1]

示例 3:

复制代码
输入:nums = [], target = 0
输出:[-1,-1]

代码(两个二分查找,自己写的更容易理解)

java 复制代码
class Solution {
    public int[] searchRange(int[] nums, int target) {
        int start = getleft(nums,target);
        int end = getright(nums,target);
        //target不存在
        if(start == -1 || end == -1){
            return new int[]{-1,-1};
        }
        //target存在
        return new int[]{start,end};  
        
    }
    //返回值start是开始边界
    public int getleft(int[] nums,int target){
        int left = 0;
        int right = nums.length - 1;
        int mid;
        int start = -1; //左边界,如果不存在,返回-1
        while(left <= right){
            mid = left + (right - left) / 2;
            //到右区间找
            if(target > nums[mid]){
                left = mid + 1;
            }
            //到左区间找
            else if(target < nums[mid]){
                right = mid -1;
            }
            //找到了目标值
            else{
                start = mid;  //start指向当前找到的mid
                //因为找开始边界,mid前面可能还有相同的target,所有right要往左走
                right = mid -1; //right要往左区间继续找,保证找到第一个target
            }
        }
        return start;  //如果没找到target,start没有被修改还是-1
    }
    //返回值end是结束边界
    public int getright(int[] nums,int target){
        int left = 0;
        int right = nums.length - 1;
        int mid;
        int end = -1; //右边界
        while(left <= right){
            mid = left + (right - left) / 2;
            if(target < nums[mid]){
                right = mid - 1;
            }
            else if(target > nums[mid]){
                left = mid + 1;
            }
            else{
                end = mid; //end指向当前找到的mid
                //因为找结束边界,mid后面可能还有相同的target,所有left要往右走
                left = mid + 1; //left要往右区间继续找,保证找到最后一个target
            }
        }
        return end; //如果没找到target,end没有被修改还是-1
    }
}

代码(两个二分查找,别人写的)

java 复制代码
class Solution {
    public int[] searchRange(int[] nums, int target) {
        int start = getleft(nums,target);
        int end = getright(nums,target);
        //target在数组的最前面和最后面
        if(start == -2 || end == -2){
            return new int[]{-1,-1};
        }
        //有target这个数,长度大于1,返回找到的区间
        if(end - start > 1){
            return new int[]{start+1,end-1};
        }
        //target在数组区间内,但找不到,长度小于等于1
        else{
            return new int[]{-1,-1};
        }
        
    }
    public int getleft(int[] nums,int target){
        int left = 0;
        int right = nums.length - 1;
        int mid;
        int start = -2; //左边界
        while(left <= right){
            mid = left + (right - left) / 2;
            if(target > nums[mid]){
                left = mid + 1;
            }
            else if(target <= nums[mid]){
                right = mid -1;
                start = right;
            }
        }
        return start;
    }
    public int getright(int[] nums,int target){
        int left = 0;
        int right = nums.length - 1;
        int mid;
        int end = -2; //右边界
        while(left <= right){
            mid = left + (right - left) / 2;
            if(target < nums[mid]){
                right = mid - 1;
            }
            else if(target >= nums[mid]){
                left = mid + 1;
                end = left;
            }
        }
        return end;
    }
}
相关推荐
爱吃生蚝的于勒5 分钟前
深入学习指针(5)!!!!!!!!!!!!!!!
c语言·开发语言·数据结构·学习·计算机网络·算法
羊小猪~~9 分钟前
数据结构C语言描述2(图文结合)--有头单链表,无头单链表(两种方法),链表反转、有序链表构建、排序等操作,考研可看
c语言·数据结构·c++·考研·算法·链表·visual studio
王哈哈^_^34 分钟前
【数据集】【YOLO】【VOC】目标检测数据集,查找数据集,yolo目标检测算法详细实战训练步骤!
人工智能·深度学习·算法·yolo·目标检测·计算机视觉·pyqt
星沁城36 分钟前
240. 搜索二维矩阵 II
java·线性代数·算法·leetcode·矩阵
脉牛杂德1 小时前
多项式加法——C语言
数据结构·c++·算法
legend_jz1 小时前
STL--哈希
c++·算法·哈希算法
kingmax542120081 小时前
初三数学,最优解问题
算法
一直学习永不止步1 小时前
LeetCode题练习与总结:赎金信--383
java·数据结构·算法·leetcode·字符串·哈希表·计数
小刘|2 小时前
《Java 实现希尔排序:原理剖析与代码详解》
java·算法·排序算法
jjyangyou2 小时前
物联网核心安全系列——物联网安全需求
物联网·算法·安全·嵌入式·产品经理·硬件·产品设计