前言
在本科毕设结束后,我开始刷卡哥的“代码随想录”,每天一节。自己的总结笔记均会放在“算法刷题-代码随想录”该专栏下,以下为“代码随想录”此题原文的链接。
35. 搜索插入位置
题目
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 1:
输入: [1,3,5,6], 5
输出: 2
示例 2:
输入: [1,3,5,6], 2
输出: 1
示例 3:
输入: [1,3,5,6], 7
输出: 4
示例 4:
输入: [1,3,5,6], 0
输出: 0
1. 判断是否使用二分查找的3个前提条件
当题目中包含数据结构为“数组”(比如链表就不可以使用二分查找,因为二分查找只能在内存地址连续的数据结构中进行),并且该“数组为有序数组”以及**“数组中无重复元素”**这三个前提条件,很有可能可以使用“二分查找”。
2. 本题思路分析:
思路:
此题目要求返回target在数组中的位置,若在数组中找不到该元素,则返回顺序插入的位置下标。
通过示例可知,顺序插入,相当于就是找“第一个大于target的数组元素所对应的下标位置”。
所以,这道题实质上就是在一个有序数组中找第一个大于等于target的下标
总共分为四种情况
情况1:target小于数组中的所有元素
情况2:target等于数组中的某元素
情况3:target大小处在数组中的元素之间(但是数组中不存在target大小的值)
情况4:target大于所有的元素
四种情况都举一个例子,如下图。
3. 算法实现
- 定义target在数组中的范围区间:通过两个int类型,left、right变量充当左右指针的值,来规定区间范围,区间的边界都闭合,区间定义为[left,right]。一个int类型(mid)变量充当中间值的下标。
区间应始终符合区间的定义,即区间大小应该大于等于1,也就是说左指针应该不大于右指针,即left <= right。 - 当数组中不含target的应对策略:这也是此题的难点,若区间范围为0时(即left=right+1,也就是left刚刚大于right时),还未找到target,则说明该区间不含有target,此时返回左区间left的下标即为题目中顺序插入的位置下标。
(为什么此是返回left下标就是题目中顺序插入的位置下标呢?
题目中顺序插入的下标实质上就是在一个有序数组中找第一个大于target的下标。又因为此时left是第一个大于target的下标(可以看上面的“本题思路分析”中列举的第一、第三、第四,这三种情况的图解,就能发现这个规律),所以返回left。
扩展:在跳出循环后,此时的right指向小于target的最后一个元素上。相当于,若target元素按照升序插入数组,正好插入到left、right之间,并且注意left>right(left=right+1) ,即left在right右边。(这个规律也可以从上方列举的第一、第三、第四,这三种情况的图解清晰地看出来) - 代码:
Java
public int searchInsert(int[] nums, int target) {
intleft = 0,right = nums.length - 1;
int cur = 0;
//左闭右闭
while(left <= right){
cur = left + ((right - left) >> 1);
if(target < nums[cur]){
right = cur - 1;
}else if(target > nums[cur]){
left = cur + 1;
}else{
return cur;
}
}
return left;
}
C++
#include<stdio.h>
#include<vector>
using namespace std;
int searchInsert(vector<int>& nums, int target) {
int left = 0,right = nums.size()-1,mid;//定义target所存在的区间,以left、right为此区间的边界,并且边界左闭右闭
while(left <= right){//当不满足区间定义时跳出循环
//左右边界都是闭区间,当左右边界在一个元素上的时候(也就是left==right),区间此时为最小区间,区间大小为1,符合区间定义
mid = left + (right - left) / 2;//left + (right - left) / 2这样防止溢出,等同于(left+right)/2
if(nums[mid] < target){//目标值存在于原数组的右半部分,新区间为[mid+1,right],左边界右移left=mid+1
left = mid + 1;
}else if(nums[mid] > target){//目标值存在于原数组左半部分,新区间为[left,mid-1],右边界左移
right = mid - 1;
}else{//此时是left、right、mid都集中在一个元素上,找到目标值返回下标
return mid;
}
}
//若跳出循环则说明数组中不含有查找的元素,
//同时意味着left,right表示的区间已经不符合区间定义,即此时区间大小变为0,
//此时left、right之间的关系必然是left=right+1,
//因为当left==right时还符合区间定义(此时区间大小为1),当left=right+1时就刚好会不满足循环条件而跳出
//此时left为下标指向的元素就是"数组中大于target的第一个元素",这就是题目所寻找的位置,故返回left
//扩展:right此时是小于target的最后一个元素
//相当于若target元素按照升序插入数组,正好插入到left、right之间,并且注意left>right(left=right+1)
return left ;
}
4. 算法分析
n为数组长度
时间复杂度:O(logn)
空间复杂度:O(1)
5. 算法的坑
- 不容易理解,当target不包含在数组中,为什么此时返回为left下标。
答:这也是此题的难点,题目中说target不存在数组中,则元素顺序插入的下标实质上就是在一个有序数组中找第一个大于target的下标。又因为此时left是第一个大于target的下标(可以看上面的“本题思路分析”中列举的第一、第三、第四,这三种情况的图解,就能发现这个规律),所以返回left。(或者返回right+1也可,此时right比left小1)