35.leetcode题目讲解(Python): 搜索插入位置

2018-11-16  本文已影响30人  夏山闻汐

题目如下:


image.png

这道题使用双指针折半查找比较容易解,指的注意的是,如果出现插入元素与输入数组里的元素相同,需要插入到相同元素的位置,比如: [8,5,3],5 应该返回1,而不是2。
参考代码如下:

class Solution:
    def searchInsert(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """

        le = len(nums)
        if le == 0:
            return 0

        if target > nums[-1]:
            return le
        elif target < nums[0]:
            return 0
        elif target = nums[-1]:
            return le - 1
        elif target = nums[0]:
            return 0

        lo = 0
        hi = le - 1

        while lo < hi:
            if (hi - lo) // 2 > 0:
                mid = lo + (hi - lo) // 2
                if nums[mid] < target:
                    lo = mid
                elif nums[mid] > target:
                    hi = mid
                elif nums[mid] == target:
                    return mid
            else:
                return hi

        return hi

源码地址:
https://github.com/jediL/LeetCodeByPython

其它题目:[leetcode题目答案讲解汇总(Python版 持续更新)]
(https://www.jianshu.com/p/60b5241ca28e)

ps:如果您有好的建议,欢迎交流 :-D,
也欢迎访问我的个人博客 苔原带 (www.tundrazone.com)

上一篇下一篇

猜你喜欢

热点阅读