题目:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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
代码:
# C++ 二分法
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int left = 0;
int right = nums.size() - 1;
while(left <= right){
int mid = (right - left) / 2 + left;
if (nums[mid] < target){
left = mid + 1;
}else if(nums[mid] > target){
right = mid - 1;
}else{
return mid;
}
}
return right + 1;
}
};
# C++ 暴力解法
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
for(int i = 0; i < nums.size(); i++){
if(target <= nums[i]){
return i;
}
}
return nums.size();
}
};
关键思路:
- 首先升序以及无重复元素。可以考虑二分法。
- 给出一个问题,要宏观看,整体上把握列出所有的情况:
- target值存在于数组中
- target值介于第一个元素和最后一个元素之间,但元素不存在
- target小于第一个元素,在数组最前
- target大于最后一个元素,在数组最后
针对暴力解法:for循环依次比较数组中的元素。首先如果是数组中存在target,return i;如果是第三种情况,第一次循环i=0时就该结束了,因而target应该插入i=0的位置,return i;如果是介于中间,nums[i]<target,循环继续,nums[i]>target时,循环结束,此时target大于i-1位置的元素,所以target插入i的位置,return i;如果target大于所有的元素,当i=len-1时,仍然没有返回值,循环结束,把target放在最后,数组增加1,return len。
针对二分算法:先用二分算法,解决target在数组元素范围内的两个情况。如果数值存在,return mid;如果不存在,left和right一直更新,直到right<left,因为如果不存在,中间一定会经历right=left,此时mid也等于right等于left,因为数值不存在,所以target应该插入在此时的数值后面,因而在跳出while条件后,right在left前面一个位置,直接return right + 1。同时,也可以发现,当target小于所有元素时,left不会动,right一直向左移动,直到right<left,也就是right=-1时,此时同样return right + 1。或大于所有元素时,right一直不动,最终的结果,很明显就是 return right + 1。
3.解决算法问题,思路清晰,先把所有情况分析出来,再逐一分析,可能会让问题简单一些,又能防止漏掉一些情况。