题目
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n)
的算法。
这道题我认为还是不那么简单的
思路
要求时间复杂度为O(logn)的话,第一个应该想到的是二分查找
注意:为什么最后返回left? 因为退出while循环后,left > right,此时有两种情况: 一种是middle指向的值大于target(此时middle是第一个大于target的下标),此时right=middle-1导致了left大于right, 而target要插入的位置是middle的位置,即left,因为left和middle相等 另一种是middle指向的值小于target,此时left=middle+1导致了left大于right,而target要插入的位置是middle+1的位置,即left
代码
class Solution {
public int searchInsert(int[] nums, int target) { // 时间复杂度要求O(logn) 使用二分查找
if (nums == null || nums.length == 0) {
return 0;
}
int left = 0,right = nums.length -1; // 代表数组最前和最后的元素
while(left<=right) {
int mid = (left + right)/2; // 如果数组长度是偶数,则mid会指向中间两个数中左边的数
if (nums[mid] == target) {
return mid;
} else if (target < nums[mid]) {
right = mid-1;
} else {
left = mid +1;
}
}
// 此时 left = right
return left;
}
}