35 搜索插入位置

本文最后更新于：2021年2月4日 上午

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

你可以假设数组中无重复元素。

示例 1:

1
2
输入: [1,3,5,6], 5
输出: 2

示例 2:

1
2
输入: [1,3,5,6], 2
输出: 1

示例 3:

1
2
输入: [1,3,5,6], 7
输出: 4

示例 4:

1
2
输入: [1,3,5,6], 0
输出: 0

Solution

  • 二分搜索
  • 定义target在左闭右闭的区间里，[left, right] ，while(l <= r) 的终止条件是 left == right + 1;
  • 定义target在左闭右开的区间里，[left, right)，while(l < r) 的终止条件是 left == right;
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
// @lc code=start
class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int l=0, r=nums.size()-1;
        while(l<=r){
            int mid = (l+r)>>1;
            if(nums[mid]==target)
                return mid;
            else if(nums[mid]>target)
                r = mid-1;
            else 
                l = mid+1;
        }
        return r+1;
    }
};
// @lc code=end
algo leetcode 算法

本博客所有文章除特别声明外，均采用 CC BY-SA 4.0 协议 ，转载请注明出处！