思路
本题用到了二分法,二分法的具体讲解在 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的位置了。