35 搜索插入位置
题目:给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 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
示例 4:
输入: nums = [1,3,5,6], target = 0
输出: 0
示例 5:
输入: nums = [1], target = 0
输出: 0
提示:
1 <= nums.length <= 104
-104 <= nums[i] <= 104
nums 为无重复元素的升序排列数组
-104 <= target <= 104
题目关键词:有序数组、无重复、O(log n)
采用方法:二分法
【大家注意这道题目的前提是数组是有序数组,这也是使用二分查找的基础条件。】
「只要看到面试题里给出的数组是有序数组,都可以想一想是否可以使用二分法。」
这题使用二分法的方式和【704二分查找】这题类似。
第一种方法:
采用区间左闭右闭的方式,即[left, right]。代码如下:
int searchInsert(vector<int>& nums, int target) {
int left = 0, right = nums.size() - 1;//定义一个左闭右闭的区间,[left, right)
while (left <= right) {
int mid = left + ((right - left) >> 1);
if (target < nums[mid]) {
right = mid - 1;
}
else if (target > nums[mid]) {
left = mid + 1;
}
else {
return mid;
}
}
return right + 1;
}
时间复杂度:O(logn)
空间复杂度:O(1)
第二种方法:
采用区间左闭右开的方式,即[left, right)。代码如下:
int searchInsert(vector<int>& nums, int target) {
int left = 0, right = nums.size();//定义一个左闭右开的区间,[left, right]
while (left < right) {
int mid = left + ((right - left) >> 1);
if (target < nums[mid]) {
right = mid;
}
else if (target > nums[mid]) {
left = mid + 1;
}
else {
return mid;//返回目标值
}
}
return right;//返回插入位置
}
时间复杂度:O(logn)
空间复杂度:O(1)
「二分查找的循环中,坚持循环不变量的原则」
确定要查找的区间到底是左闭右开[left, right),还是左闭又闭[left, right],这就是不变量。