1.题目
链接:https://leetcode-cn.com/leetbook/read/array-and-string/cxqdh/
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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] <= 104
nums 为 无重复元素 的 升序 排列数组
-104 <= target <= 104
2. 解题
1.顺序遍历
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int index = 0;
for (int i = 0; i < nums.size(); i++){
if (target < nums[i] || nums[i] == target) return i;
if (target > nums[i]) {
index = i + 1;
}
}
return index;
}
};
2.二分查找
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int n = nums.size();
int left = 0, right = n - 1, ans = n;
while (left <= right) {
int mid = ((right - left) >> 1) + left; // ">>" 向右位移操作,相当于除以2
if (target <= nums[mid]) {
ans = mid;
right = mid - 1;
} else {
left = mid + 1;
}
}
return ans;
}
};
链接:https://leetcode-cn.com/problems/search-insert-position/solution/sou-suo-cha-ru-wei-zhi-by-leetcode-solution/
3.递归
编写递归代码时的注意事项:
- 递归总有一个最简单的情况,方法的第一条语句总是一个包含return的条件语句;
- 递归调用总是去尝试解决一个规模更小的子问题,这样递归才能够收敛到最简单的情况,在下面的代码中,第四个参数和第三个参数的差值一直在缩小;
- 递归调用的负问题和尝试解决的子问题之间不应该会有交集。在下面的代码中,两个子问题各自操作数组的不同部分。
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int right = nums.size() - 1;
int left = 0;
return searchInsert(nums, target, left, right);
}
int searchInsert(vector<int>& nums, int target, int left, int right){
if (left > right) return left;
int mid = ((right - left) >> 1) + left;
if (target == nums[mid]) return mid;
else if (target > nums[mid]) return searchInsert(nums, target, mid + 1, right);
else return searchInsert(nums, target, left, mid - 1);
}
};