给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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提示:
1 <= nums.length <= 104-104 <= nums[i] <= 104nums为 无重复元素 的 升序 排列数组-104 <= target <= 104
解题思路 -- 求左边界端点
这道题就可以使用上我们所学的模板啦,我们先来找找规律!
因为我们如果找不到元素的话,需要插入元素,并且是在找不到元素的位置的右端插入元素,也就是在 target 左侧的第一个大于等于 target 的元素位置。
而如果找到了元素的话,那么就是返回最左端的那个 target,那题目就转化为了求左边界的问题,直接套模板!
不过这里有细节问题,比如题目中的示例三,如果 target 比数组元素都大,在求左边界的时候找不到 target 的话,那么 left 指针是会一直向后走的,但是它只会最后停在数组末尾,但是找不到元素的话,此时是需要返回末尾后面那个位置,所以需要特判一下!
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
// 相当于查找左边界
int left = 0, right = nums.size() - 1;
while(left < right)
{
int mid = left + (right - left) / 2;
if(nums[mid] < target)
left = mid + 1;
else
right = mid;
}
if(nums[left] < target) // 特判
return left + 1;
return left;
}
};