给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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
提示:
1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 为无重复元素的升序排列数组
-104 <= target <= 104
由于数组有序,查找targe很容易想到二分查找,即折半查找。
首先,假设表中元素是按升序排列,将表中间位置记录的关键字与查找关键字比较,如果两者相等,则查找成功;否则利用中间位置记录将表分成前、后两个子表,如果中间位置记录的关键字大于查找关键字,则进一步查找前一子表,否则进一步查找后一子表。重复以上过程,直到找到满足条件的记录,使查找成功,或直到子表不存在为止,此时查找不成功。
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
/*左侧指针*/
int low = 0;
/*右侧指针*/
int high = nums.size() - 1;
/*当左侧指针小于等于右侧指针时遍历*/
while (low <= high) {
/*计算中间节点*/
int mid = (low + high) / 2;
/*当中间节点大于目标值时,右侧节点移到中间节点的左边一位*/
if(nums[mid] > target) {
high = mid - 1;
/*当中间节点小于目标值时,左侧节点移到中间节点的右边一位*/
}else if (nums[mid] < target) {
low = mid + 1;
}else{
/*找到了返回目标值当前所在位置*/
return mid;
}
}
/*返回左侧节点,即是要插入的位置*/
return low;
}
};
测试数据:
执行用时:
0 ms, 在所有 C++ 提交中击败了100.00%的用户
内存消耗:
9.4 MB, 在所有 C++ 提交中击败了46.45%的用户
通过测试用例:
64 / 64