题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。 你可以假设数组中无重复元素。
示例 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
寻找插入位置,最直接的想法是遍历数组,找到与target值相等的元素的下标,然后直接返回,或者,在数组中找到比target值小,但是下一个元素比target值大的元素,返回对应下标。
另一想法就是直接使用二分查找法
以下代码为二分查找法代码:
class Solution {
public int searchInsert(int[] nums, int target) {
//初始化左、右指针
int left = 0, right = nums.length - 1;
while (left < right) {
int mid = left + (right - left) / 2;
//根据中间元素与target的大小比较来更新左右指针
if (target > nums[mid]) {
left = mid + 1;
} else if (target < nums[mid]) {
right = mid - 1;
} else {
//如果target==nums[mid],直接返回mid即可
return mid;
}
}
//如果没有相同元素,则target应该在left的左或者右
return target > nums[left] ? left + 1 : left;
}
}