LeetCode——35. 搜索插入位置(二分)

35. 搜索插入位置

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目

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

你可以假设数组中无重复元素。

示例 1:

输入: [1,3,5,6], 5
输出: 2
示例 2:

输入: [1,3,5,6], 2
输出: 1
示例 3:

输入: [1,3,5,6], 7
输出: 4
示例 4:

输入: [1,3,5,6], 0
输出: 0

思路

二分查找来查找下标
1、注意特殊情况当目标target大于数组所有数据,则返回数组长度;
2、根据二分查找来剪枝,即当数组数据小于target时不是解,进行区间舍弃。left = mid + 1;数组值等于target时返回下标即可;当大于target时返回第一个大于target的元素下标即可,mid=right;
3、最终left与right相会,返回left下标即可

代码

class Solution {
    public int searchInsert(int[] nums, int target) {
        int len = nums.length;
        if(nums[len - 1] < target){//特殊判断,目标值大于数组所有值
            return len;
        }
        return binarySearch(nums, 0, len - 1, target);
    }
    public int binarySearch(int[] nums, int left, int right, int target) {
        while(left < right){
            int mid = left + (right - left) / 2;
            if(nums[mid] == target){
                return mid;
            }
            else if(nums[mid] < target){//数组值小于target时不是解
                left = mid + 1;
            }
            else{//数组值严格大于target的是解
                right = mid;//注意不是mid - 1,默认下取整;上取整是left = mid; right = mid - 1;
            }
        }
        return left;
    }
}

猜你喜欢

转载自blog.csdn.net/qq_34767784/article/details/107021163