给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
解析:
这是一个二分查找的变形,看到有序数组,基本就想到二分查找。这个题目可以用二分查找,找到元素的相应位置,如果没有目标元素,则找到刚好比它大的位置,就是要插入的位置,但是有一个特殊情况,就是插入的元素比所有元素都大,此时元素应该插在数组中最后一个元素的后一个位置。
代码:
int searchInsert(vector<int>& nums, int target)
{
int start = 0;
int end = nums.size() - 1;
if (target > nums[end])//比最后一个元素大
return end + 1;
while (start < end)
{
int mid = (end - start) / 2 + start;
if (nums[mid] > target)//目标值小于mid位置的值
{
end = mid;
}
else if (nums[mid] == target)//目标值等于mid位置的值
{
return mid;
}
else//目标值大于mid位置的值
{
start = mid + 1;
}
}
return start;
}