【LeetCode】35.搜索插入位置

【LeetCode】35.搜索插入位置

思路

本题用到了二分法，二分法的具体讲解在 704.二分查找 中已经展示过了，可以先做完二分查找再来做本题。

代码

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int left = 0;
        int right = nums.size() - 1;
        int mid = 0;
        //1.二分法在数组nums中找target
        while (left <= right)
        {
            mid = (left + right) / 2;
            if (target < nums[mid])
            {
                right = mid - 1;
            }
            else if (target > nums[mid])
            {
                left = mid + 1;
            }
            else
            {
                return mid;
            }
        }
        //2.没有在数组中找到target，需要返回target的插入位置
        if (target > nums[mid])
        {
            return mid + 1;
        }
        else
        {
            return mid;
        }
    }
};

注意

本题相对二分查找需要额外注意的就是，当要查找元素不在数组中，需要返回要插入的位置时就需要比较一下nums[mid]与target的大小了。

以下图为例：

由于target = 10不在数组中，循环结束后mid = 6，此时target < nums[6]，故target应插入在下标为mid的位置。

若target = 16，那么 target就需要插入在mid + 1的位置了。