给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
思路,二分法搜索,找到相等返回索引,不等则返回应该插入的索引。难度在于边界问题的处理。
class Solution {
public int searchInsert(int[] nums, int target) {
int len=nums.length,left=0,right=len-1,mid=0;
if(len==0) return 0;
if(nums[0]>=target) return 0; //加黑代码为判定边界条件 也可以不写这两行
if(nums[len-1]<target) return len; //切记不能用<=
else if(nums[len-1]==target) return len-1;
while(left<=right)
{
mid=(left+right)/2;
if(nums[mid]==target)
return mid;
else if(nums[mid]<target)
left=mid+1;
else if(nums[mid]>target)
right=mid-1;
}
return left;
}
}
附上AC成功图:
PS:加黑的3行代码也可以不加,它们主要用于判定边界情况。不过最好加上,避免边界情况降低二分法搜索的时间效率。
特别是LeetCode,边界情况特别多.......