题目
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
解答
class Solution {
public:
int searchInsert(vector<int>& nums, int target)
{
int left = 0, right = nums.size()- 1;
while(left <= right)
{
int mid = (left + right) >> 1;
if(nums[mid] == target)
return mid;
else if(nums[mid] < target)
left = mid + 1;
else
right = mid - 1;
}
return left;
}
};
这题是接着上两题二分查找做的,不同之处在于return的值。题目中说的是如果找不到返回下一个。当跳出while循环的时候left肯定是在right右边一个,此时正好是需要插入的位置