标签:二分查找 pre 初始化 重复元素 初始 相等 src return 位置
联系二分查找
Q:给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。你可以假设数组中无重复元素。
示例 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
A:
复习二分查找。
但实际上有更方便的做法。
因为我们初始化 right = nums.length,所以决定了我们的「搜索区间」是 [left, right),所以决定了 while (left < right)
同时也决定了 left = mid + 1 (left不可取mid)和 right = mid(right可取mid)
因为我们需找到 target 的最左侧索引,所以当 nums[mid] == target 时不要立即返回,而要收紧右侧边界以锁定左侧边界,最后相等时left=right,取[left,right)。
代码:
public int searchInsert(int[] nums, int target) {
if (nums.length == 0)
return -1;
int left = 0, right = nums.length;
while (left < right) {//搜索区间是 [left, right)
int mid = (left + right) / 2;
if (nums[mid] == target)
right = mid;//收紧右侧边界以锁定左侧边界
else if (nums[mid] < target)
left = mid + 1;
else
right = mid;
}
return left;
}
标签:二分查找 pre 初始化 重复元素 初始 相等 src return 位置
原文地址:https://www.cnblogs.com/xym4869/p/12603389.html