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

题目 搜索插入位置

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

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

猜你喜欢

转载自blog.csdn.net/new_DYX/article/details/107400228