Leetcode--34--Search for a Range

题型：binary search

关键在于二分查找中left，right，mid三者以及num[mid]与target的大小关系设置。

 

举例：nums = [1, 2, 2, 6, 7]         target = 2 或者 3

根据这个样例可以得出以下具体结果

 

首先，分4种情况：

1. 如果target存在nums中，得到>=target的最左位置

2. 如果target存在nums中，得到>=target的最右位置

3. 如果target不存在nums中，得到>target的最左位置

4. 如果target不存在nums中，得到>target的最右位置

 

#解决情况1, 4
#情况1：因为nums[mid]==target，r位置会向左移动
#情况4：因为return l，返回的l是如果对于target小于nums[mid]来说，l是会+1；否则r会-1,；不管怎样都是取的>target的位置
l = 0
r = len(nums) - 1
while l <= r:
    mid = l + (r - l) / 2
    if nums[mid] == target:
        r = mid - 1
    elif nums[mid] > target:
        r = mid - 1
    else:
        l = mid + 1
return l

 

#解决情况2, 3
#理由同上
l = 0
r = len(nums) - 1
while l <= r:
        mid = l + (r - l) / 2
        if nums[mid] == target:
            l = mid + 1
        elif nums[mid] > target:
            r = mid - 1
        else:
            l = mid + 1

 

 

solution code：

class Solution(object):
    def searchRange(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        if len(nums)==0:
            return [-1,-1]
        if nums[len(nums)-1] < target or nums[0] > target:
            return [-1,-1]
        l = 0
        r = len(nums) - 1
        # <= 
        while l <= r:
            mid = l + (r - l) / 2
            if nums[mid] == target:
                r = mid - 1
            elif nums[mid] > target:
                r = mid - 1
            else:
                l = mid + 1
        left = l
        l = 0
        r = len(nums) - 1
        while l <= r:
            mid = l + (r - l) / 2
            if nums[mid] == target:
                l = mid + 1
            elif nums[mid] > target:
                r = mid - 1
            else:
                l = mid + 1
        right = r
        if nums[left] == target:
            return [left, right]
        else:
            return [-1, -1]
            #return [right, left]