给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
示例 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
思路:
第一种:
二分查找:
如果该题目暴力解决的话需要 O(n)O(n) 的时间复杂度,但是如果二分的话则可以降低到 O(logn)O(logn) 的时间复杂度
整体思路和普通的二分查找几乎没有区别,先设定左侧下标 left 和右侧下标 right,再计算中间下标 mid
每次根据 nums[mid] 和 target 之间的大小进行判断,相等则直接返回下标,nums[mid] < target 则 left 右移,nums[mid] > target 则 right 左移
查找结束如果没有相等值则返回 left,该值为插入位置
时间复杂度:O(logn)O(logn)
class Solution {
public int searchInsert(int[] nums, int target) {
int left = 0, right = nums.length - 1; int a=0;
while(left <= right) {
int mid = (left + right) / 2;
if(nums[mid] == target) {
//相关逻辑
a=mid;
} else if(nums[mid] < target) {
left = mid + 1;
} else {
right = mid - 1;
}
}
// 相关返回值
return a;
}
}
第二种:
这个解法也是有一些取巧
class Solution {
public int searchInsert(int[] nums, int target) {
int a=nums.length;
for(int i=0;i<nums.length;i++){
if(nums[i] > target){
a=i;
return a;
}
if(nums[i] == target){
a=i;
return a;
}
}
return a;
}
}