LeetCode P35--搜索插入位置

题目来源:力扣(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

寻找插入位置,最直接的想法是遍历数组,找到与target值相等的元素的下标,然后直接返回,或者,在数组中找到比target值小,但是下一个元素比target值大的元素,返回对应下标。

另一想法就是直接使用二分查找
以下代码为二分查找法代码:

class Solution {
    
    
    public int searchInsert(int[] nums, int target) {
    
    
    	//初始化左、右指针
        int left = 0, right = nums.length - 1;
        while (left < right) {
    
    
            int mid = left + (right - left) / 2;
            //根据中间元素与target的大小比较来更新左右指针
            if (target > nums[mid]) {
    
    
                left = mid + 1;
            } else if (target < nums[mid]) {
    
    
                right = mid - 1;
            } else {
    
    
            //如果target==nums[mid],直接返回mid即可
                return mid;
            }
        }
        //如果没有相同元素,则target应该在left的左或者右
        return target > nums[left] ? left + 1 : left;
    }
}

猜你喜欢

转载自blog.csdn.net/weixin_44215175/article/details/107416760