搜索插入位置

题目描述

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

请必须使用时间复杂度为 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⁴
• -10⁴ <= nums[i] <= 10⁴
• nums 为 无重复元素 的 升序 排列数组
• -10⁴ <= target <= 10⁴

二分查找

// Solution 类的定义
class Solution {
public:// searchInsert 函数采用两个参数：一个整数类型的 vector 引用 nums 和一个整数 targetint searchInsert(vector<int>& nums, int target) {// 如果目标值小于数组中的第一个元素，则它应该被插入在数组的开头if(target < nums[0]) return 0;// 如果目标值大于数组中的最后一个元素，则它应该被插入在数组的末尾if(target > nums[nums.size() - 1]) return nums.size();// 初始化二分查找的左边界 l 和右边界 rint l = 0, r = nums.size(); // 注意这里右边界是开区间，即 r 初始值等于数组长度while(l < r) // 当左边界小于右边界时，执行循环{// 计算中点 mid，这里使用位运算右移一位来快速计算平均值int mid = (l + r) >> 1;// 如果中点的元素大于等于目标值，则说明目标值位于左侧或就是 mid 本身if(nums[mid] >= target) r = mid; // 将右边界移动到中点else l = mid + 1; // 否则，目标值位于右侧，移动左边界}// 最终，左边界 l 将指向目标值的插入位置return l;}
};

暴力

class Solution {
public:int searchInsert(vector<int>& nums, int target) {for (int i = 0; i < nums.size(); i++) {// 分别处理如下三种情况// 目标值在数组所有元素之前// 目标值等于数组中某一个元素// 目标值插入数组中的位置if (nums[i] >= target) { // 一旦发现大于或者等于target的num[i]，那么i就是我们要的结果return i;}}// 目标值在数组所有元素之后的情况return nums.size(); // 如果target是最大的，或者 nums为空，则返回nums的长度}
};