文章目录
35. 搜索插入位置
来源:力扣(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
思路
二分查找来查找下标
1、注意特殊情况当目标target大于数组所有数据,则返回数组长度;
2、根据二分查找来剪枝,即当数组数据小于target时不是解,进行区间舍弃。left = mid + 1;数组值等于target时返回下标即可;当大于target时返回第一个大于target的元素下标即可,mid=right;
3、最终left与right相会,返回left下标即可
代码
class Solution {
public int searchInsert(int[] nums, int target) {
int len = nums.length;
if(nums[len - 1] < target){//特殊判断,目标值大于数组所有值
return len;
}
return binarySearch(nums, 0, len - 1, target);
}
public int binarySearch(int[] nums, int left, int right, int target) {
while(left < right){
int mid = left + (right - left) / 2;
if(nums[mid] == target){
return mid;
}
else if(nums[mid] < target){//数组值小于target时不是解
left = mid + 1;
}
else{//数组值严格大于target的是解
right = mid;//注意不是mid - 1,默认下取整;上取整是left = mid; right = mid - 1;
}
}
return left;
}
}