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

    技术2022-07-10  103

    Description

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

    你的算法时间复杂度必须是 O(log n) 级别。

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

    示例 1: 输入: nums = [5,7,7,8,8,10], target = 8 输出: [3,4] 示例 2: 输入: nums = [5,7,7,8,8,10], target = 6 输出: [-1,-1] 来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

    Solution

    针对target-1和target都进行二分。

    class Solution: def searchRange(self, nums: List[int], target: int) -> List[int]: if len(nums)==0: return [-1, -1] def binary_search(target): l, r = 0, len(nums) while l < r: m = l + (r - l) //2 if nums[m] <= target: l = m + 1 else: r = m return l l = binary_search(target-1) r = binary_search(target) if l > r-1: return [-1, -1] else: return [l, r-1]
    Processed: 0.008, SQL: 9