题目描述
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
思路:二分法
- 如果该题目暴力解决的话需要 O(n)的时间复杂度,但是如果二分的话则可以降低到 O(logn)的时间复杂度
- 先设定左侧下标 left 和右侧下标 right,再计算中间下标 mid
- 每次根据 nums[mid] 和 target 之间的大小进行判断,相等则直接返回下标,nums[mid] < target 则 left右移,nums[mid] > target 则 right 左移
- 查找结束如果没有相等值则返回 left,该值为插入位置
- 二分查找的思路不难理解,但是边界条件容易出错,比如 循环结束条件中 left 和 right 的关系,更新 left 和 right 位置时要不要加 1 减 1。
模板
模板记住就好,免除边界条件出错。
class Solution { public int searchInsert(int[] nums, int target) { int left = 0, right = nums.length - 1; // 注意 while(left <= right) { // 注意 int mid = (left + right) / 2; // 注意 if(nums[mid] == target) { // 注意 // 相关逻辑 } else if(nums[mid] < target) { left = mid + 1; // 注意 } else { right = mid - 1; // 注意 } } // 相关返回值 return 0; } }
代码实现
class Solution { public int searchInsert(int[] nums, int target) { //定义左侧下标left和右侧下标right int left = 0, right = nums.length - 1; while (left <= right) { int mid = (left + right) / 2; //当nums[mid]和目标元素相等时,直接返回 if (nums[mid] == target) { return mid; //当nums[mid]小于目标元素时,left右移 [mid+1,right]为下一个区间 } else if (nums[mid] < target) { left = mid + 1; //反之,当nums[mid]大于目标元素时,right左移 [left,mid-1]为下一个区间 } else { right = mid - 1; } } //查找结束如果没有相等值则返回 left,该值为插入位置 return left; } }
原文地址:https://www.cnblogs.com/xiaozhongfeixiang/p/12439277.html
时间: 2024-10-09 19:09:02