[leetcode/lintcode 题解] 算法面试真题详解:在排序数组中查找元素的第一个和最后一个位置-阿里云开发者社区

开发者社区> 算法编程> 正文
登录阅读全文

[leetcode/lintcode 题解] 算法面试真题详解:在排序数组中查找元素的第一个和最后一个位置

简介: [leetcode/lintcode 题解] 算法面试真题详解:在排序数组中查找元素的第一个和最后一个位置

描述
给定一个按照升序排列的整数数组 nums,和一个目标值 target。找出给定目标值在数组中的开始位置和结束位置。
你的算法时间复杂度必须是 O(log n) 级别。
如果数组中不存在目标值,返回 [-1, -1]。

在线评测地址:[领扣题库官网](https://www.lintcode.com/problem/1536/?utm_source=sc-tianchi
-sz-0514)

样例1
输入: nums = [5,7,7,8,8,10], target = 8
输出: [3,4]
样例2
输入: nums = [5,7,7,8,8,10], target = 6
输出: [-1,-1]

解题思路
不用烧脑, 直接两次无脑二分,求left 和 right。
不过需要注意的是
if numsmid <= target: start = mid else: end = mid 得到是right, 我第一次以为是left。 感觉如果再碰到,还是有可能写这个bug出来,究竟怎么想会好一点呢?大家有什么方法?

源代码

class Solution:
    def searchRange(self, nums, target):
        if not nums:
            return [-1, -1]
        
        left, right = -1, -1
        start, end = 0, len(nums) - 1 
        while start + 1 < end:
            mid = (start + end) // 2 
            if nums[mid] <= target:
                start = mid
            else:
                end = mid
        if nums[start] == target:
            right = start
        if nums[end] == target:
            right = end
        
        start, end = 0, len(nums) - 1 
        while start + 1 < end:
            mid = (start + end) // 2 
            if nums[mid] >= target:
                end = mid
            else:
                start = mid
        if nums[end] == target:
            left = end
        if nums[start] == target:
            left = start
        if left == -1 and right == -1:
            return [-1, -1]
        return [left, right]

更多题解参考:九章官网solution

版权声明:本文内容由阿里云实名注册用户自发贡献,版权归原作者所有,阿里云开发者社区不拥有其著作权,亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容,填写侵权投诉表单进行举报,一经查实,本社区将立刻删除涉嫌侵权内容。

分享: