- 题目需求
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。示例 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
- 解法
1:遍历数组,从下标为0开始向后遍历,当值大于等于目标值,返回下标值,时间复杂度O(n)
public int searchInsert(int[] nums, int target) {
if (nums.length == 0) {
return 0;
}
for (int i = 0; i < nums.length; i++) {
if (nums[i] >= target)
return i;
}
return nums.length;
}
2:二分搜索,时间复杂度O(log N)
public int binarySearchInsert(int[] nums, int target) {
if (nums.length == 0) return 0;
int l = 0, r = nums.length - 1;
while (l <= r) {
int m = l + ((r - l) >> 1);
if (target == nums[m] || ((m == 0 || target > nums[m - 1]) && target < nums[m])) return m;
if (target > nums[m]) l = m + 1;
else r = m - 1;
}
return nums.length;
}