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

问题分析

这道题当然可以直接遍历，另一种思路是采取二分法，这里解释下为什么最后返回left,以实例2为例，开始left = 0,right = 3,middle = 1,nums[1] = 3 >2,因此right = middle =1,这时新的middle = 0,nums[0]=1<2,left = 0+1=1,这时left==right,这时left恰好在该插入的位置。

代码

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

结果：

执行用时 :8 ms, 在所有 cpp 提交中击败了71.18%的用户
内存消耗 :8.9 MB, 在所有 cpp 提交中击败了81.04%的用户

//二分查找
class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int i,n=nums.size(),right = n-1,left=0,middle;
        if(nums[right]< target) return n; 
        while(right - left > 0)
        {
            middle = left + (right - left)/2;
            if(nums[middle] < target)
            {
                left = middle + 1; 
            }
            else if(nums[middle] > target)
            {
                right = middle;
            }
            else{
                return middle;
            }
        }
        
        return left;

    }
};

结果：

执行用时 :4 ms, 在所有 cpp 提交中击败了97.48%的用户
内存消耗 :8.9 MB, 在所有 cpp 提交中击败了74.45%的用户