前言
做题练习。
练习八 【35.搜索插入位置】
一、题目阅读
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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 <= 10^4
-10^4 <= nums[i] <= 10^4
nums 为 无重复元素 的 升序 排列数组
-10^4 <= target <= 10^4
二、尝试实现
思路
题目找target,还要求时间复杂度Q(log n);并且nums排好序,升序。所以想到二分查找,双指针法。
(1)二分查找的底子,判断相等,直接return middle;代表找到。
(2)如果nums[middle] < target,更新left位置。更新之后,判断此时nums[left] > target,那么target不在数组内,可以return 插入位置。
(3)如果nums[middle] > target,更新right位置。更新之后,判断此时nums[right] < target,那么target不在数组内,可以return 插入位置。
(4)以上考虑情况: nums[0] < target < nums[nums.size()-1],也就是target在数组范围内。如果target超过nums最大值,小于nums最小值,上面的逻辑不覆盖。单独处理。
代码实现
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
if(target > nums[nums.size()-1]){
return nums.size();
}else if(target < nums[0]){
return 0;
}
//在中间
int left = 0;
int right = nums.size()-1;
int middle = (left+right)/2;
while(left <= right){
if(nums[middle] == target){ //如果找到
return middle;
}else if(nums[middle] < target){ //不相等,
left = middle + 1;
if(left < nums.size() && nums[left] > target) return left;
}else if(nums[middle] > target){
right = middle -1;
if(right >=0 && nums[right] < target) return middle;
}
middle = (left+right)/2 ;//更新middle
}
return -1;//走不到这,所以给个-1
}
};
三、参考学习
学习内容
- 注意边界处理。四种情况:
- 目标值在数组所有元素之前
- 目标值等于数组中某一个元素
- 目标值插入数组中的位置
- 目标值在数组所有元素之后
-
暴力解法:遍历数组。
class Solution { public: int searchInsert(vector<int>& nums, int target) { for(int i = 0;i < nums.size();i++){ if(nums[i] >= target){ return i; } } return nums.size(); } };
3.二分查找:重点——循环不变量。保持区间开闭情况始终一致。
- 习惯使用左闭右闭区间来写二分查找。参考处理不同之处:
- 如果找到,直接return middle;
- 如果没找到,target在数组之前,此时right退出循环时等于-1,所以插入位置0就等于 right+1;
- 如果target在数组之后,此时right退出循环时等于nums.size()-1,所以插入位置nums.size()就等于right+1;
- 如果target在数组中间,此时right退出循环时nums[right] < target并且是比target小的元素中的最大值,left > right,说明left把小于target的元素跨过去。依然是right+1.
- 统一了找不到的情况。都是return right+1.
- 代码实现
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int left = 0;
int right = nums.size()-1;
int middle = (left+right)/2;
while(left <= right){
if(nums[middle] == target){ //如果找到
return middle;
}else if(nums[middle] < target){ //不相等,
left = middle + 1;
}else if(nums[middle] > target){
right = middle -1;
}
middle = (left+right)/2 ;//更新middle
}
return right+1;
}
};
总结
【35.搜索插入位置】
(欢迎指正,转载标明出处)