问题描述
分析
题目要求我们必须使用O(logn)的算法,其实这不就是给我们提示么。看到这个时间复杂度我们自然而然地想到二分查找,若查找到对应的元素则返回对应下标,若没查找到对应的元素应该返回target应该插入的位置。后者我们可以分成两种情况来讨论,第一种:如果target小于最后一趟二分查找nums[mid]的值则返回mid;第二种:如果target大于最后一趟二分查找nums[mid]的值则返回mid + 1。
代码实现
public int searchInsert(int[] nums, int target) {
int low = 0,high = nums.length - 1,mid = 0;
//二分查找中如果low指针大于high指针了,便是要查找的目标元素不存在待查数组中
while (low <= high) {
mid = (low + high) / 2;
if (nums[mid] == target)
return mid;
else if (target < nums[mid])
high = mid - 1;
else if (target > nums[mid])
low = mid + 1;
}
return target > nums[mid] ? mid + 1 : mid;
}