给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 1:
输入: [1,3,5,6], 5 输出: 2
示例 2:
输入: [1,3,5,6], 2 输出: 1
示例 3:
输入: [1,3,5,6], 7 输出: 4
示例 4:
输入: [1,3,5,6], 0 输出: 0
思路:
没什么难度,思路有两种,一种直接遍历暴力找到位置,最优时间复杂度O(1),最坏O(n),第二种用二分法,最坏时间复杂度为O(logn),最优O(1)。
代码:
暴力法:
class Solution {
public int searchInsert(int[] nums, int target) {
int i = nums.length;
for(i = 0;i < nums.length;i++){
if(target <= nums[i]){
return i;
}
}
return i;
}
}
二分法:
class Solution {
public int searchInsert(int[] nums, int target) {
int left = 0;
int right = nums.length - 1;
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] == target) return mid;
else if (nums[mid] > target) right = mid - 1;
else left = mid + 1;
}
return left;
}
}