题目:给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
分析:
标签:二分查找
如果该题目暴力解决的话需要 O(n)O(n)O(n) 的时间复杂度,但是如果二分的话则可以降低到 O(logn)O(logn)O(logn) 的时间复杂度
整体思路和普通的二分查找几乎没有区别,先设定左侧下标 left 和右侧下标 right,再计算中间下标 mid,每次根据 nums[mid] 和 target 之间的大小进行判断,相等则直接返回下标,nums[mid] < target 则 left 右移,nums[mid] > target 则 right 左移,查找结束如果没有相等值则返回 left,该值为插入位置, 时间复杂度:O(logn)O(logn)O(logn)
二分查找的思路不难理解,但是边界条件容易出错,比如 循环结束条件中 left 和 right 的关系,更新 left 和 right 位置时要不要加 1 减 1。
代码:
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
// 定义target在左闭右开的区间里,[left, right) target
int left=0,right=nums.size();
// 因为left == right的时候,在[left, right)是无效的空间
while(left<right){
int mid=left+(right-left)/2;
// target 在左区间,在[left, middle)中
if(nums[mid]>target){
right=mid;
// target 在右区间,在 [middle+1, right)中
}else if(nums[mid]<target){
left=mid+1;
}else{
return mid; // 数组中找到目标值的情况,直接返回下标
}
}
return left;
}
};