LeetCode Search Insert Position

LeetCode解题之Search Insert Position

---

原题

在一个有序数组中，如果目标数字存在，则返回它的下标，否则返回它应该插入位置的下标值。

注意点：

• 数组中没有重复数字

例子：

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

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

解题思路

又是一道二分搜索的题。分以下几种情况：如果当前数字是目标数字，或者当前数字大于目标数字而它之前的数字小于目标数字，则返回当前下标；如果当前数字为最后一个且它比目标数字小，则返回当前下标的后一位。

AC源码

class Solution(object):    def searchInsert(self, nums, target):        """        :type nums: List[int]        :type target: int        :rtype: int        """        length = len(nums)        start = 0        end = length        while start < end:            mid = (start + end) // 2            if nums[mid] == target or (nums[mid] > target                 and (mid == 0 or nums[mid - 1] < target)):                return mid            if mid == length - 1 and nums[mid] < target:                return mid + 1            if nums[mid] < target:                start = mid + 1            else:                end = midif __name__ == "__main__":    assert Solution().searchInsert([1, 3, 5, 6], 5) == 2    assert Solution().searchInsert([1, 3, 5, 6], 2) == 1    assert Solution().searchInsert([1, 3, 5, 6], 7) == 4    assert Solution().searchInsert([1, 3, 5, 6], 0) == 0

欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。