给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
示例1
输入: nums = [1,3,5,6], target = 5
输出: 2
示例2
输入: nums = [1,3,5,6], target = 2
输出: 1
示例3
输入: nums = [1,3,5,6], target = 7
输出: 4
示例4
输入: nums = [1,3,5,6], target = 0
输出: 0
示例5
输入: nums = [1], target = 0
输出: 0
这道题使用二分法解决比较简单,大家如果看到题目里面给出的数组是有序数组,都可以考虑一下是否可以使用二分法。
答案(Java)
class Solution {
public int searchInsert(int[] nums, int target) {
//判断目标值是否存在数组中
//使用二分法查找target是否在数组中
int left=0,right=nums.length-1;
while(left<=right){
int mid=left+(right-left)/2;//为了避免溢出,等同于(left+right)/2
if(target==nums[mid]){
//target在数组中,返回索引
return mid;
}else if(target<nums[mid]){
right=mid-1;
}else{
left=mid+1;
}
}
//关于数组中没有目标值,有三种可能要考虑清楚
//target小于第一个数
if(target<nums[0])
return 0;
//target大于最后一个数
if(target>nums[nums.length-1])
return nums.length;
//target不在数组中,但是范围是在数组中
int i=0;
while(nums[i]<target){
++i;
}
return i;
}
}
通过了,但是内存消耗很大,不是理想答案
数组中没有目标值的这一情况中,可以简化代码,直接返回right+1也是可以的
除了二分法,还可以直接暴力求解
class Solution {
public int searchInsert(int[] nums, int target) {
//暴力求解
for(int i=0;i<nums.length;++i){
//target在数组前,在数组里,是数组中的某个元素
if(target<=nums[i])
return i;
}
//target的值大于数组中的任何数
return nums.length;
}
}