难度
简单
地址
https://leetcode.cn/problems/search-insert-position/description/
题目
给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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
思路
-
二分查找
如果该题目暴力解决的话需要 O(n) 的时间复杂度，但是题目中明确要求时间复杂需要是O(log n)，故采用二分的话则可以降低到 O(logn) 的时间复杂度
步骤：
先设定左侧下标 l 和右侧下标 r，再计算中间下标 mid
每次根据 nums[mid] 和 target 之间的大小进行判断，相等则直接返回下标，nums[mid] < target 则 l 右移，nums[mid] > target 则 r 左移
查找结束如果没有相等值则返回 l，该值为插入位置
class Solution {
public int searchInsert(int[] nums, int target) {
int l = 0;
int r = nums.length - 1;
while (l <= r) {
int mid = (l + r) / 2;
if(nums[mid] == target){
return mid;
}else if(nums[mid] < target){
l = mid+1;
}else {
r = mid -1;
}
}
return l;
}
}
二分查找的模版代码（通用）
class Solution {
public int searchInsert(int[] nums, int target) {
int left = 0, right = nums.length - 1; // 注意
while(left <= right) { // 注意
int mid = (left + right) / 2; // 注意
if(nums[mid] == target) { // 注意
// 相关逻辑
} else if(nums[mid] < target) {
left = mid + 1; // 注意
} else {
right = mid - 1; // 注意
}
}
// 相关返回值
return 0;
}
}