这道题主要有四种情况分别是:
1.目标值在数组所有元素之前
2.目标值等于数组中某一个元素
3.目标值插入数组中的位置
4.目标值在数组所有元素之后
我们采取二分法求解这道
举个例子:找到数值为5的数组元素下标
以下的代码中定义 target 是在一个在左闭右闭的区间里,也就是[left, right]
class Solution {
public://左闭右闭方式[1,3,5,6],target = 5
int searchInsert(vector<int>& nums, int target) {
int left = 0;
int right = nums.size()-1;//right = 3左闭右闭区间为[0, 3]
while(left<=right)// 当left==right,区间[left, right]依然有效
{
int middle = left+((right-left)>>1);// 防止溢出 等同于(left + right)/2
if(nums[middle]>target)
{
right = middle-1;// target 在左区间,所以[left, middle - 1]
}
else if(nums[middle]<target)
{
left = middle+1;// target 在右区间,所以[middle + 1, right]
}
else{// nums[middle] == target
return middle;
}
}
return right+1;
}
};
// 分别处理如下四种情况
// 目标值在数组所有元素之前 [0, -1]
// 目标值等于数组中某一个元素 return middle;
// 目标值插入数组中的位置 [left, right],return right + 1
// 目标值在数组所有元素之后的情况 [left, right], 因为是右闭区间,所以 return right + 1
- 时间复杂度:O(log n)
- 空间复杂度:O(1)