leetcode刷题——35. 搜索插入位置

原创

顾道长生 2022-07-13 09:54:46 博主文章分类：OJ刷题 ©著作权

文章标签 leetcode 算法数组时间复杂度空间复杂度 文章分类 后端开发

©著作权归作者所有：来自51CTO博客作者顾道长生的原创作品，请联系作者获取转载授权，否则将追究法律责任

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

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

示例 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 flag=0;
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]>=target)
            {
                return i;
            }
        }
        return nums.size();
    }
};

Note

vector m;
m.size();
我一开始写成了length（），编译错误

用二分法解决

代码

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int left=0;
        int right=nums.size()-1;
        while(left<=right)
        {
            int middle=left+((right-left)/2);//防止溢出
            if(nums[middle]>target)
                right=middle-1;
            else if(nums[middle]<target)
                left=middle+1;
            else
                return middle;
        }
        return right+1;
    }
};