leetcode学习：34.在排序数组中查找元素的第一个和最后一个位置

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

题目描述

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

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

进阶：

你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗？

示例

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

思路

其实还是一个二分查找，只不过对查找出来的位置进行往前往后的处理。

代码

class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        def find(index, nums, target):
            res = [index, index]
            for i in range(index - 1, -1, -1):
                    if nums[i] == target:
                        res[0] = i
                    else:
                        break
            for i in range(index + 1, len(nums)):
                if nums[i] == target:
                    res[1] = i
                else:
                    break
            return res
        l = 0
        r = len(nums) - 1
        mid = 0
        while l <= r:
            if nums[l] == target:
                return find(l, nums, target)
            if nums[r] == target:
                return find(r, nums, target)
            mid = (l + r) // 2
            if nums[mid] == target:
                return find(mid, nums, target)
            if nums[mid] > target:
                r = mid - 1
            else:
                l = mid + 1
        return [-1, -1]  

运行结果

执行用时：
32 ms, 在所有 Python3 提交中击败了96.85%的用户
内存消耗：
15.9 MB, 在所有 Python3 提交中击败了8.53%的用户

题目来源

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。