LeetCode搜索插入位置求解(35题)(Python和C++实现)

一、题目描述：

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

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

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

 

二、解决办法

1. 解决办法一：

采用暴力求解的方式，算法复杂度O(N) （Python实现）。

class Solution(object):
    def searchInsert(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        '''
        # 暴力求解，复杂度O(n)
        n = len(nums)
        for i in range(n):
            if nums[i] >= target:
                return i 
        return n
        '''

2. 解决办法二：

采用二分法进行求解，算法复杂度Olog(N)。其中二分法根据开闭区间，提供两种思路：

思路一：左右两侧都是闭区间 （Python实现）。

   def searchInsert(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        '''       
        #二分查找
        n = len(nums)
        left = 0
        right = n-1
        while (left <= right):
            mid = int((left + right)/2)
            if nums[mid] > target:
                right = mid-1
            elif nums[mid] < target:
                left = mid+1
            else :
                return mid
        return right + 1

思路二：左侧闭区间，右侧开区间 （C++实现）。

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int n = nums.size();
        int left = 0;
        int right = n - 1; // 定义target在左闭右闭的区间里，[left, right] 
        while (left <= right) { // 当left==right，区间[left, right]依然有效
            int middle = left + ((right - left) / 2);// 防止溢出 等同于(left + right)/2
            if (nums[middle] > target) {
                right = middle - 1; // target 在左区间，所以[left, middle - 1]
            } else if (nums[middle] < target) {
                left = middle + 1; // target 在右区间，所以[middle + 1, right]
            } else { // nums[middle] == target
                return middle;
            }
        }
        // 分别处理如下四种情况
        // 目标值在数组所有元素之前  [0, -1]
        // 目标值等于数组中某一个元素  return middle;
        // 目标值插入数组中的位置 [left, right]，return  right + 1
        // 目标值在数组所有元素之后的情况 [left, right]， return right + 1
        return right + 1;
    }
};

 

三、算法比较

下图的第一个通过是二分法求解，第二个通过是暴力求解。可以看出在这道题目中，二分法查找和暴力求解的运行时间差不多，二分法的内存消耗比暴力的循环少一点。

1. 暴力求解的具体时间和空间复杂度;

2. 二分法的具体时间和空间复杂度：

 

参考文献：

1. https://mp.weixin.qq.com/s/fCf5QbPDtE6SSlZ1yh_q8Q

2. https://www.bilibili.com/video/BV187411d77t?from=search&seid=4957648913905534017