Leetcode35-搜索插入位置详解

往期博客：

Leetcode1-两数之和详解

Leetcode2-两数相加代码详解

Leetcode20-有效的括号详解

Leetcode21-合并两个有序链表详解

Leetcode22-有效括号生成详解

Leetcode24-两两交换链表中的节点详解

Leetcode27-移除元素详解

---

目录

题目

示例

解析

常规法

 二分法

代码

常规法

python

Java

二分法

python

Java

---

题目

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

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

题目分析

已知：排序数组和一个目标值

目的：在数组中找到目标值，或按顺序插入目标值

要求：返回目标值的索引

---

示例

示例1

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

示例2

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

示例3

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

---

解析

常规法

对于给定的数组nums，从头遍历所有元素，直到第一个大于等于val的元素 i 出现，如果 i 等于target则说明数组中存在数值等于target的元素，此时直接返回 i 的索引，如果 i 大于target，则说明数组中不存在数值等于target的元素，则将target插入 i 的位置，并返回 i 的索引，但是如果数组遍历到最后时仍小于target，说明素组中所有元素都小于target，则将target插入到数组最后，即索引值为数组长度的位置。

看具体示例

对于示例1，数组为nums=[1,3,5,6]，target = 5

首先从头遍历数组，直到遇到第一个等于5的元素时停止遍历，返回5的索引值2

对于示例2，数组nums=[1,3,5,6]，target = 2

首先从头遍历数组，直到遇到第一个大于2的元素3时停止遍历

 将target插入到此位置，并返回索引值1

 对于示例3，数组nums=[1,3,5,6]，target = 7

首先从头遍历数组，当遍历到数组最后时仍小于target，此时将target输入到数组最后，target的索引值为原数组的长度

---

 二分法

对于给定的数组nums，创建左右指针，左指针left指向数组头部，右指针right指向数组尾部，则中间指针mid指向左右指针的中间位置，然后判断中间的元素a和中target，如果a=target，则直接返回中间指针，如果a>target，则移动右指针到mid+1位置处，如果a<target，则移动左指针到mid位置处，当左指针大于等于右指针时退出循环，并返回左指针。

看具体示例

对于示例1，数组为nums=[1,3,5,6]，target = 5

创建指针

 由于nums[mid] < target，移动左指针到mid+1位置处，此时nums[mid]和target相等，则直接返回mid

对于示例2，数组nums=[1,3,5,6]，target = 2

 创建指针

 由于nums[mid] > target，移动右指针到中间指针位置处，此时mid指向1

 由于nums[mid] < target，移动左指针到mid+1位置处，此时左右指针相等，退出循环，返回左指针

---

代码

常规法

python

方法一

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        
        # 当数组为空时，直接返回0
        if len(nums) == 0 or nums == None:
            return 0

        for i, num in enumerate(nums):
            if num >= target:
                return if
        
        return len(nums)

方法二

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        
        # 当数组为空时，直接返回0
        if len(nums) == 0 or nums == None:
            return 0

        prev = -1
        for i in range(len(nums)):
            if nums[i] == target:
                return i
            elif nums[i] < target:
                prev = i
            else:
                return prev + 1
        return prev + 1

Java

class Solution {
    public int searchInsert(int[] nums, int target) {
        if (nums == null || nums.length == 0){
            return 0;
        }

        for (int i = 0; i < nums.length; i++){
            if (nums[i] >= target){
                return i;
            }
        }
        return nums.length;

    }
}

二分法

python

class Solution:
    def searchInsert(self, nums: List[int], target: int) -> int:
        
        # 当数组为空时，直接返回0
        if len(nums) == 0 or nums == None:
            return 0

        left = 0
        right = len(nums)

        while left < right:
            mid = left + (right - left)//2
            if nums[mid] == target:
                return mid
            elif nums[mid] < target:
                left = mid + 1
            else:
                right = mid

        return left

Java

class Solution {
    public int searchInsert(int[] nums, int target) {
        if (nums == null || nums.length == 0){
            return 0;
        }

        int left = 0;
        int right = nums.length - 1;

        while(left < right){
            int mid = left + (right - left) / 2;
            if (nums[mid] == target){
                return mid;
            }else if (nums[mid] > target){
                right = mid;
            }else{
                left = mid + 1;
            }
        }

        return nums[left] < target? left + 1 : left;

    }
}