题目 搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-insert-position
思路
看看题目发现就是个典型的二分查找问题而已,因为存在比数组最大值还大的情况,会返回当前不存在的索引值,这种情况注意一下就好。顺便这应该是二分查找的内容吧,但我怎么不记得二分是不是这么处理最大元素的呢。
代码实现
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int left = 0, right = nums.size() - 1;
while (left < right - 1) {
int middle = (left + right) / 2;
if (nums[middle] == target) {
return middle;
}
nums[middle] > target ? right = middle : left = middle;
}
if (target <= nums[left]) {
return left;
}
else if (target <= nums[right]) {
return right;
}
else {
return right + 1;
}
}
};
但是我先这么写的时间结果不是很理想,那么参考下官解优化一下:
最好的优化是中间的一行代码:
int middle = ((right - left) >> 1) + left;
在取中值的时候,将(left + right) / 2转变为如上的运行方式,也就是采用左移操作去代替除以2的操作。
事实上这句代码是下面这句代码的升级版:
int middle = ((right + left) >> 1);
((right - left) >> 1) + left这个操作最好的一点是避免了right和left相加过程中的溢出问题。保证操作数不会大于right而造成溢出。
那么最终的代码如下:
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int left = 0, right = nums.size() - 1;
while (left < right - 1) {
int middle = ((right - left) >> 1) + left;
if (nums[middle] == target) {
return middle;
}
nums[middle] > target ? right = middle : left = middle;
}
if (target <= nums[left]) {
return left;
}
else if (target <= nums[right]) {
return right;
}
else {
return right + 1;
}
}
};