leetcode—数组—搜索插入位置

搜索插入位置

题目类型： 数组
题目来源： leetcode传送门

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

请必须使用时间复杂度为 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
输出: 4

示例4：

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

示例5：

输入: nums = [1], target = 0
输出: 0

提示：

• 1 <= nums.length <= 10⁴
• -10⁴ <= nums[i] <= 10⁴
• nums 为无重复元素的升序排列数组
• -10⁴ <= target <= 10⁴

同样又是一个二分的问题，简单说一下思路吧，目的是找到从左往右第一个比target大的数字，mid取的是(left+right+1)/2，采用的是向上取整的方法，因为想找的是第一个比target大的数字，那么就需要最后左指针等于mid+1，刚好和右指针重合，所以这就要采用向上取整的方法。最后一个特殊判断，如果恰好最后得到的位置是nums.size()-1，这时如果nums[nums.size()-1]<target，则返回nums.size()即可，否则其他均返回left的值，但是当发现nums[mid]等于target，这时直接返回mid就行了。

代码如下：

int searchInsert(vector<int>& nums, int target) {
        if(nums.size() == 0)
            return 0;
        int left = 0;
        int right = nums.size() - 1;
        while(left < right)
        {
            int mid = (left + right) / 2;
            if(nums[mid] < target)
            {
                left = mid + 1;
            }
            else if(nums[mid] > target)
            {
                right = mid;
            }
            else
            {
                return mid;
            }
        }
        if(left == nums.size() - 1 && nums[nums.size()-1] < target)
            return left + 1;
        return left;
    }

为了熟悉二分的写法，再来一种向下取整的方法（mid=(left+right)/2)

int searchInsert(vector<int>& nums, int target) {
        if(nums.size() == 0)
            return 0;
        int left = 0;
        int right = nums.size() - 1;
        while(left < right)
        {
            int mid = (left + right) / 2;
            if(nums[mid] < target)
            {
                left = mid + 1;
            }
            else if(nums[mid] > target)
            {
                right = mid;
            }
            else
            {
                return mid;
            }
        }
        if(left == nums.size() - 1 && nums[nums.size() - 1] < target)
        {
            return left + 1;
        }
        return left;
    }

这题其实还可以找从左向右最后一个比target小的位置，总体思路大致相同，可以自己尝试写一下。