给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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
示例1描述的场景是数组中存在某个元素的值等于目标值,那么该元素所在的索引即目标位置。
示例2描述的场景是数组中不存在某个元素的值等于目标值,那么势必可以确定某个元素的值是小于目标值,该元素的下一个元素是大于目标值,则该元素的下一个元素索引值即目标位置。
示例3描述的场景是数组中最大值也小于目标值,那么数组最大下标的下一个位置即目标位置。
示例4描述的场景是数组中最小值也大于目标值,那么数组第一个下标位置即目标位置。
public int searchInsert(int[] nums, int target) {
int length = nums.length;
if (nums[0] > target) {
return 0;
}
if (nums[length-1] < target) {
return length;
}
int left = 0;
for(int i = 0; i < length; i++) {
if (nums[i] == target) {
return i;
}
if (nums[i] < target) {
left = i;
}
}
return left + 1;
}
而且因为数组时有序的,可以利用二分查找特性:
public int searchInsert(int[] nums, int target) {
int length = nums.length;
if (nums[0] > target) {
return 0;
}
if (nums[length-1] < target) {
return length;
}
int left = 0;
int right = length - 1;
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] == target) {
return mid;
}
if (nums[mid] < target) {
left = mid + 1;
} else {
right = mid - 1;
}
}
return left;
}