Skip to content

0035.搜索插入位置

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

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

思路

使用二分查找法，在每次比较时，如果中间元素小于目标值，则将搜索范围缩小到右半部分，否则缩小到左半部分。最终，当搜索范围为空时，left 指针所指的位置即为目标值应插入的位置。

解答

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int l=0,r=nums.size();
        while(l<r){
            int mid = (l+r)/2;
            if(nums[mid] >= target) r = mid;
            else l = mid+1;
        }
        return l;
    }
};