题目难度:
Easy
题目要求:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
示例 1:
输入: nums = [1,3,5,6], target = 5
输出: 2
算法思路:
1、要求是时间复杂度为O(log n),则想到二分法;
2、考虑特殊情况,若目标值应该插入数组最右端时,ans=nums.length;若插入最左端,即ans=0;
3、根据二分法,需要知道left(最左)、right(最右)、和mid(中间)值;
4、每次,将目标值与mid作比较,如果目标值大,则说明应该插入右边区域,所以,left变为mid+1,right不变,mid重新计算再比较;如果目标值小,则应该插入左边区域,right变为mid-1,left不变,mid重新计算再比较;如果相等,则将mid赋值给ans,并返回。
代码:
class Solution {
public int searchInsert(int[] nums, int target) {
int length = nums.length;
int ans = length;
int left = 0;
int right = length - 1;
while (left <= right) {
int mid = ((right - left) >> 1) + left;
if (target <= nums[mid]) {
ans = mid;
right = mid - 1;
} else {
left = mid + 1;
}
}
return ans;
}
}
时间复杂度:O(logn);
空间复杂度:O(1);