Search Insert Position

最新推荐文章于 2023-05-31 00:05:53 发布

原创最新推荐文章于 2023-05-31 00:05:53 发布 · 224 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #array #easy

python 同时被 3 个专栏收录

20 篇文章

订阅专栏

Leetcode

12 篇文章

订阅专栏

array

4 篇文章

订阅专栏

本文介绍了一种使用二分查找算法解决特定问题的方法：给定一个已排序的数组及目标值，返回目标值在数组中的索引位置。若目标值不在数组中，则返回其应该插入的位置索引。通过对比逐个遍历的方法，文章展示了二分查找算法不仅能够有效地解决问题，而且大大提高了查找效率。

Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.

You may assume no duplicates in the array.

Here are few examples.
[1,3,5,6], 5 → 2
[1,3,5,6], 2 → 1
[1,3,5,6], 7 → 4
[1,3,5,6], 0 → 0

我比较笨的就从头到尾依次的比较，发现这并不是一个很好的办法，虽然可以完成要求，但是所花费的时间很多。
看到好的方法是用二分查找来解决这个问题。

class Solution(object):
    def searchInsert(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        if target < nums[0]:
            return 0
        elif target > nums[-1]:
            return len(nums)

        l, r = 0, len(nums) - 1
        while l <= r:
            m = (l + r)/2
            if nums[m] == target:
                return m
            elif nums[m] > target:
                if nums[m-1] < target:
                    return m
                else:
                    r = m -1
            elif nums[m] < target:
                if nums[m+1] > target:
                    return m+1
                else:
                    l = m + 1