给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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
这个题虽然在leeetcode上被标为简单题,我认为难度较之昨天的那一道增加了不少,对循环结束的条件做出了重点考察,下面给出代码:
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int len = nums.size();
int start = 0,end = len-1;
int temp;
while(start <= end)
{
temp = start + (end - start)/2;
if(nums[temp]==target)
return temp;
else if(nums[temp] < target)
start = temp + 1;
else if(nums[temp] > target)
end = temp - 1;
}
return start;
}
};
主体循环部分和上题保持一致,这里不作说明,这边解释一下为什么最后插入的部分一定是start:
首先,到后面这个循环说明原数组中不存在该元素,由于我们设置的循环条件是start<=end,所以此时start=end+1=temp+1,因为temp反映的是最后一个判断的区间,即[a,a],start=end的时候,当这个区间不满足时,执行下一步,start=temp+1或end=temp-1,无论何种情况,end都一定比start小一,而我们这时候可以根据temp的值是否等于start,分为两种情况:
1.start=temp,这说明上一步执行的是end=temp-1,此时的情况是nums[temp] > target,这时候我们应该吧target插入temp的位置,即start,满足要求;
2.start=temp+1,这说明上一步执行的是start=temp+1,此时的情况是nums[temp] <target,这时候我们应该吧target插入temp+1的位置,同样也是start的位置。