leetcode_35. 搜索插入位置

发布时间 2023-08-27 11:08:48作者: 晓枫的春天

题目描述

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

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

输入: nums = [1,3,5,6], target = 5
输出: 2
输入: nums = [1,3,5,6], target = 2
输出: 1
输入: nums = [1,3,5,6], target = 7
输出: 4
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

二分实现

和之前二分不同之处是多了一个元素可能不在列表中,需要按序插入返回对应位置,

Java 实现

    public int searchInsert(int[] nums, int target) {
        if (nums == null || nums.length == 0) {
            return 0;
        } else {
            int left = 0, right = nums.length - 1;
            while (left < right) {
                int mid = (right + left) / 2;
                if (nums[mid] == target) {
                    return mid;
                } else if (nums[mid] > target) {
                    right = mid;
                } else {
                    left = mid + 1;
                }

            }
            return left;
        }
    }

Python实现

def searchInsert1(nums: List[int], target: int) -> int:
    left, right = 0, len(nums)
    while left < right:
        mid = (left + right) / 2
        if nums[mid] == target:
            return mid
        elif nums[mid] > target:
            right = mid
        elif nums[mid] < target:
            left = mid + 1

    return left

暴力法

Java 实现 

    public int searchInsert_baoli(int[] nums, int target) {
        if (nums == null || nums.length == 0) {
            return 0;
        } else {
            for (int i = 0; i < nums.length; i++) {
                if (nums[i] >= target) {
                    return i;
                }
            }
            return nums.length;
        }
    }

Python 实现

def searchInsert_baoli(nums: List[int], target: int) -> int:
    if nums is None or len(nums) == 0:
        #判断是否需要及逆行运算
        return 0
    else:
        for i in range(len(nums)):
            #找到第一个大于等于目标值的索引-返回
            if nums[i] >= target:
                return i
        #最后一个位置:即数组大小
        return len(nums)