原题目链接: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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
做题思路一(暴力查询):
- 由于数组是有序的,直接遍历数组,与 target 值作比较
1.1 如果第一个的元素就大于 target 值,那 target 值一定放在0下标,因为数组是有序的,左边就是最小的,右边是最大的,如果找到遍历到元素比 target 值大,就返回该索引
1.2 如果等于 target 值,就直接返回索引
1.3 如果遍历完没有以上的发现,就说明 target 值比数组的数据都大,直接放在数组末尾
废话不多说,直接上代码,为了让各位看官更能清晰理解,我的代码写得不精简,我的代码里加了大量的注释,相信各位看官可以理解,如果我有些没写清楚或者写错的,可以评论区或者私信我喔
public int searchInsert(int[] nums, int target) {
//遍历数组
for(int i = 0; i < nums.length; i++){
//因为是有序的数组,所以0下标所指的元素肯定是最小的
if(nums[i] > target) return i;
//如果找到目标值,就返回索引
if(nums[i] == target) return i;
}
//如果数组找不到目标值,就把目标值添加在数组末尾
return nums.length;
}
做题思路二(二分法):
- 我们先了解一下什么是二分法,二分法就是一种查找数据的方法,但是前提是,这组数据一定是有序的。简单来说就是不断地取中间
- 我们以 [1,3,5,6], 5 为例,首先准备一个 left 指针和 right 指针,left 的指针的索引是0,right 的指针的索引是3,然后不断地取中间的 mid 指针
2.1 第一次 (left + right)/ 2 也就是 (0 + 3)/ 2,就是 mid 指针的值 1
2.2 此时我们发现 mid 指针指的元素是小于目标元素5,因为假设以 mid 为中心,左边的数肯定是小于 mid 右边的数肯定是大于等于 mid ,所以此时我们想要查找的数据肯定在右边
2.3 所以我们就把 left 移动到 mid ,然后再进行一次 2.1 的操作,(1 + 3)/ 2 ,就是 mid 的值 2
2.4 此时我们发现 mid 指针指的元素是等于目标元素5,所以 mid 指针的位置,也就是索引2,就是我们要查找的位置,这就是二分查找法
废话不多说,直接上代码,为了让各位看官更能清晰理解,我的代码写得不精简,我的代码里加了大量的注释,相信各位看官可以理解,如果我有些没写清楚或者写错的,可以评论区或者私信我喔
public int searchInsert(int[] nums, int target) {
//数据已经排好序
//左边最小数据的索引,右边是最大数据的索引
int left = 0;int right = nums.length - 1;
//比最小还小,那就直接放在最左
if(nums[left] > target) return 0;
//比最大还大,那就直接放在最右
if(nums[right] < target) return nums.length;
//左闭右开的二分法
//当左边大于等于右边的时候,说明二分法已经对整个数组进行了查找
while(left < right){
//取中间值
// >>1 是代表除以2的意思,这里写成位运算效率比较高,当然也可以写成 / 2
int mid = (left + right) >> 1;
if(nums[mid] >= target){ //如果元素大于等于目标值,说明要往小的范围去查
//因为是排好序了,右边一定大于左边,所以把右边的界限缩小
right = mid;
}else{
//如果元素小于目标值,说明要往大的范围去查
//因为是排好序了,右边一定大于左边,所以把左边的界限缩小
//由于是左闭右开,所以这里要 + 1
left = mid + 1;
}
}
//这里返回left或者right都可以,因为题目说了,数组没有重复元素
return left;
}