一、题目
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为
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
二、分析
根据题目要求,是要用时间复杂度为log n的算法,二分查找的算法的时间复杂度为log n,而且这也是一个有序的数组,我们通过mid的值就可以知道了,一半数字是比target大还是小,这里我们找的是第一个 >= target的元素的位置,代码入下;
class Solution {
public int searchInsert(int[] nums, int target) {
//使用时间复杂度为log n 的算法,所以要用二分查找
int n = nums.length ;
int l = 0;
int r = n-1;//闭区间
while(l <= r){
int mid = l + (r-l)/2;//这个等同于(r+l)/2,这样写是为了防止整数溢出,当r和l都非常大的时候,相加可能超过整数的范围
// if(mid <target){
// l = mid + 1;
// }else{
// r = mid -1;
// }
if(nums[mid] < target){
l = mid + 1;
}else{
r = mid -1;
}
}
return l; //也可以返回 r+1
}
}
这里很粗心的一点,把mid当作了一个值,它是数组的下标 ,可以参照视频
还有第二种方法就是暴力遍历
class Solution {
public int searchInsert(int[] nums, int target) {
int a =0;
if(target > nums[nums.length-1]) return nums.length;
if(target == nums[nums.length-1]) return nums.length-1;
if(target < nums[0]) return 0;
for(int i = 0; i < nums.length-1; i++){
if(nums[i] == target){
a = i;
}else if(nums[i] < target && nums[i+1] > target){
a = i+1;
}
}
return a;
}
}
在写暴力的方法时,我在for循环里判断的时候用到了nums[i+1],如果i的取值范围时第一个元素到最后一个元素的话,那么会有数组越界的问题,所以范围应该是第一个元素到倒数第一个元素,这个之后,所以最后一个元素没有被考虑到,所以单独判断一下;
除此之外,还要想一下如果有比第一个元素还小的数,那么插入的位置会是0.
如果有任何问题请欢迎指出,或者有更好的思路。