在排序数组中查找元素的第一个和最后一个位置 Python

一、题目描述

给你一个按照非递减顺序排列的整数数组 nums，和一个目标值 target。请你找出给定目标值在数组中的开始位置和结束位置。

如果数组中不存在目标值 target，返回 [-1, -1]。

示例 1

输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]
1
2

示例 2

输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]
1
2

示例 3

输入：nums = [], target = 0
输出：[-1,-1]
1
2

提示：
0 <= nums.length <= 105
-109 <= nums[i] <= 109
nums 是一个非递减数组
-109 <= target <= 109

二、代码

代码如下：

class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
		# 返回情况2.2 2.3
        if not nums or nums[0] > target or nums[-1] < target:
            return [-1, -1]
        else:    # 返回情况 1， 2.1    
            left_border = self.binary_search_left_border(nums, target, 0, len(nums)-1)
            right_border = self. binary_search_right_border(nums, target, 0, len(nums)-1)
            return [left_border, right_border]
        

    def binary_search_left_border(self, nums, target, left, right) -> int:
        while left <= right:
            mid = left + (right - left) // 2
            if nums[mid] == target:
                right = mid - 1 # 向左逼近
            elif nums[mid] > target:
                right = mid - 1
            else:
                left = mid + 1
        if nums[left] == target:
            return left
        else:
            return -1

    def binary_search_right_border(self, nums, target, left, right) -> int:
        while left <= right:
            mid = left + (right - left) // 2
            if nums[mid] == target:
                left = mid + 1 # 向右逼近
            elif nums[mid] < target:
                left = mid + 1
            else:
                right = mid - 1
        if nums[right] == target:
            return right
        else:
            return -1

三、解题思路

本题的思路是在二分查找的基础上在找到位置处向左右进行查找，如果左右仍然存在与target目标值相同的数，则记录下他们的下标到index1中，最终的index1中存放有所有与target目标值相同的元素，此时排序过后输出index1[0]和index1[len(index1)-1]就是重复元素出现的第一个位置和最后一个位置。