【每日算法】LeetCode 35 —— 搜索插入位置 (一百一二十七)

题目内容

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

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

示例

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

题解

本题在求解上思路很简单直接,就是使用二分算法,针对target,将区间分为两部分,求出下标。

这里需要强调一下边界情况,由于存在数组中的数全部小于target的情况,target的位置需要放置在数组末尾,因此右边界的下标范围需要设置为nums.size()。

代码

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

return l;
}
};
Author: Frederic Niu
Link: https://www.fredericniu.cn/2021/05/01/【每日算法】LeetCode-35-——-搜索插入位置-(一百一二十七)/
Copyright Notice: All articles in this blog are licensed under CC BY-NC-SA 4.0 unless stating additionally.
我的公众号