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

本文阅读 2 分钟
首页 代码,Java 正文

给定一个按照升序排列的整数数组 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]

 

使用两个二分查找,第一个二分找第一个出现的位置,第二个二分找最后一个出现的位置。即使使用了两个二分查找,时间复杂度仍为 log(n).

步骤:

①找第一个出现的位置,当 arr[mid] == target && target > arr[mid-1] 说明 mid 就是第一个出现的位置,否则递归调用,不断向前缩小范围,最后考虑边界,即 mid == 0,此时如果 arr[mid] == target 就返回 0,否则返回 -1.

②找最后一个出现的位置,当 arr[mid] == target && target < arr[mid + 1] 时,说明 mid 就是最后一个出现的位置,否则递归调用,不断向后缩小范围,最后考虑边界情况,即 mid == end 时,此时如果 arr[mid] == target 就返回 end,否则返回 -1.  

class Solution(object):
    def searchRange(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: List[int]
        """
        first = self.searchFirst(nums, 0, len(nums) - 1, target)
        last = self.searchLast(nums, 0, len(nums) - 1, target)
        return [first, last]
    
    
    def searchFirst(self, nums, start, end, target):
        if end >= start:
            mid = int((start + end) / 2)
            if nums[mid] == target and (mid == 0 or target > nums[mid - 1]):
                return mid
            elif nums[mid] < target:
                return self.searchFirst(nums, mid + 1, end, target)
            else:
                return self.searchFirst(nums, start, mid - 1, target)
        
        return -1
    
    def searchLast(self, nums, start, end, target):
        if end >= start:
            mid = int((start + end) / 2)
            print(mid)
            if nums[mid] == target and (mid == end or target < nums[mid + 1]):
                return mid
            elif nums[mid] <= target:   # 注意与搜索第一个数的区别
                return self.searchLast(nums, mid + 1, end, target)
            else:
                return self.searchLast(nums, start, mid - 1, target)

        return -1

 

本文为互联网自动采集或经作者授权后发布,本文观点不代表立场,若侵权下架请联系我们删帖处理!文章出自:https://wangjiawei.blog.csdn.net/article/details/102786525
-- 展开阅读全文 --
安全面试之XSS(跨站脚本攻击)
« 上一篇 07-24

发表评论

成为第一个评论的人

热门文章

标签TAG

最近回复