[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

上一篇:[leetcode/lintcode 题解]算法面试真题详解:二叉搜索树结点最小距离


下一篇:[leetcode/lintcode 题解] 阿里巴巴面试题: 路径和 IV