欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

35. 搜索插入位置

程序员文章站 2022-03-13 12:57:23
...

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

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

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

思路一:循环遍历,返回第一个大于等于目标值的数的下标,否则返回数组大小

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        for(int i=0;i<=nums.size();++i)
        {
            if(nums[i]>=target)
                return i;
           
        }
        return nums.size();
    }
};

思路二:运用二分法查找

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

 

相关标签: 二分法