每日一解 简单问题 搜索插入位置

题目 搜索插入位置

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。

示例 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;
		}
	}
};