题目描述:
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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 <= nums.length <= 10^4
-10^4 <= nums[i] <= 10^4
nums
为 无重复元素 的 升序 排列数组-10^4 <= target <= 10^4
解法一:利用for循环进行遍历,如果nums[i]大于等于target,返回i,否则返回nums.length
代码如下:
public static int searchInsert(int[] nums, int target) {
for (int i = 0; i < nums.length; i++) {
if (nums[i] >= target){
return i;
}
}
return nums.length;
}
解法二:利用二分法
代码如下:
public static int searchInsert(int[] nums, int target) {
//定义left和right的初始值,声明mid
int left = 0;
int right = nums.length-1;
int mid;
//先考虑边界,nums是无重复的升序数组,如果target比数组中的第0位置上的数都小返回0
if(target<nums[0]){
return 0;
}
//right有意义,所以left可以等于right
while (left <= right) {
//不断更新mid,并向上取整,防止mid-1<0(>>是除以2的意思)
mid = (int) Math.ceil(left + ((right - left) >> 1));
if (nums[mid] == target) {
return mid;
}else if (nums[mid] < target ) {
left = mid + 1;
}else if(nums[mid] > target){
if(nums[mid-1]< target){
return mid;
}
right = mid-1;
}
}
return nums.length;
}