35. 搜索插入位置

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

必须使用时间复杂度为 O(log n) 的算法。

  示例 1:

    输入: nums = [1,3,5,6], target = 5
    输出: 2
  示例 2:

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

===============================================================

这个题目也是经典的二分法,唯一麻烦的就是最后的判定条件,上代码

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

给一个官方的代码:

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

最后炫耀一下:

                                                    

 

posted on 2022-01-05 17:27  4小旧  阅读(28)  评论(0)    收藏  举报

导航