Search Insert Position
Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.
You may assume no duplicates in the array.
Example 1:
Input: [1,3,5,6], 5
Output: 2
Example 2:
Input: [1,3,5,6], 2
Output: 1
Example 3:
Input: [1,3,5,6], 7
Output: 4
Example 4:
Input: [1,3,5,6], 0
Output: 0
题目要求
- 给定一个有序的数组和一个目标值,如果目标值在数组中则返回这个值在数组中的下标,如果这个值不在数组中,返回这个值应该插入到有序数组中的位置。
解题思路
因为是有序的数组,所以首选的方法是二分查找。查找的逻辑如果,取数组的首尾下表(左闭右开)以及中间位置middle((left+right)/2),如果中间位置的值恰好是target,则返回找到的这个下标。
再判断是不是边界条件,即是不是中间值已处于最左边或者最右边。在判断是不是插入的位置,插入位置的特征是middle值比target小,middle+1位置的值大于target,这样就应该插入到middle+1的位置。
最后在判断是继续向左搜索还是向右搜索。
时间复杂度O(log n),超过100%的java提交,代码如下:
class Solution {
public int searchInsert(int[] nums, int target) {
// 使用二分查找
int left = 0,right=nums.length;
int middle,result;
while(left < right){
middle=(left+right)/2;
// 处理边界情况
if(nums[middle]==target){
return middle;
}else if(middle == left && nums[left] > target) return middle>0?middle-1:0;
else if(middle == right-1 && nums[right-1] < target) return right;
else if(nums[middle]<target && nums[middle+1]>target){
return middle+1;
}else if(target < nums[middle]){
right = middle;
}else if(target >= nums[middle+1]){//这里的等于是为了处理查找时的情况
left = middle+1;
}
}
return 0;
}
}