Search for a Range

此题有个假二分做法,也有个真二分做法,假二分是找到目标元素后左右扩展找边界,这样最坏情况就是O(n)了,真二分则做两次二分,一次找左边,一次找右边,下面只写真二分

class Solution(object):
    def searchRange(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        if not nums:
            return [-1, -1]

        start, end = 0, len(nums) - 1
        while start + 1 < end:
            mid = start + (end - start) / 2
            if nums[mid] <= target:
                start = mid 
            else:
                end = mid

        left, right = 0, len(nums) - 1
        if nums[end] == target:
            right = end
        elif nums[start] == target:
            right = start
        else:
            return [-1, -1]

        start, end = 0, len(nums) - 1
        while start + 1 < end:
            mid = start + (end - start) / 2
            if nums[mid] >= target:
                end = mid
            else:
                start = mid

        if nums[start] == target:
            left = start
        elif nums[end] == target:
            left = end
        else:
            return [-1, -1]
        return [left, right]

results matching ""

    No results matching ""