给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n)
的算法。
代码如下所示:
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int left = 0;
int right = nums.size() - 1;
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] == target) {
return mid;
} else if (nums[mid] < target) {
left = mid + 1;
} else {
right = mid - 1;
}
}
return left;
}
};
int left = 0;
和int right = nums.size() - 1;
:初始化左右指针,分别指向数组的头部和尾部。while (left <= right)
:这个循环会持续进行,直到找到目标值或确定目标值的插入位置。int mid = left + (right - left) / 2;
:计算中间位置。使用left + (right - left) / 2
来避免left + right
可能出现的溢出问题。if (nums[mid] == target)
:如果mid
位置的元素等于目标值,直接返回该索引。else if (nums[mid] < target)
:如果mid
位置的元素小于目标值,目标值应该在mid
右侧,更新left = mid + 1
。else
:如果mid
位置的元素大于目标值,目标值应该在mid
左侧,更新right = mid - 1
。return left;
:如果循环结束时没有找到目标值,left
将指向目标值应该插入的位置。