给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。(来源:力扣(LeetCode))
你可以假设数组中无重复元素。
示例 1:
输入: [1,3,5,6], 5
输出: 2
思路:要在数组中插入目标值,需要判断该值在数组中出现的位置(四种):
- 目标值在所有数组元素之前
- 目标值与某个数组元素相等
- 目标值在两个数组元素之间
- 目标值在所有数组元素之后
解法:
- 普通方法
//时间复杂度为O(n),空间复杂度为O(1)
class Solution {
public int searchInsert(int[] nums, int target) {
for(int i=0;i<nums.length;i++){
//目标值小于等于数组元素值,即为上述前三种情况,返回当前索引
if(nums[i]>=target){
return i;
}
}
//第四种情况,插入位置就是数组最大长度
return nums.length;
}
}
- 二分法(有序数组,应该考虑到使用二分法,使用二分法要注意自己选择的区间)
//时间复杂度为O(logn),空间复杂度为O(1)
class Solution {
public int searchInsert(int[] nums, int target) {
int begin=0;
int end=nums.length-1;
int mid=0;
//区间为[begin,end],也可采用[begin,end(end=nums.length)],过程一样。
while(begin<=end){
mid=(begin+end)/2;
if(nums[mid]<target){
begin=mid+1;
}else if(nums[mid]>target){
end=mid-1;
}else{
//当目标值在数组元素之间和与数组元素相等时
return mid;
}
}
//目标值在所有元素之前(end=-1)和之后时(end=nums.length-1)
return end+1;
}
}
参考公众号:代码随想录的leetcode刷题最强指南