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

题目描述:

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

解题思路:

其实这道题解题思路有蛮多,我当时想的就是用列表的index()函数找到第一个target的索引,存到一个list里面,然后在原来列表中改变这个值,然后再去找下一个target,循环往复,之后再去取新建的list的第一个和最后一个,但是我这个方法有bug,过不了测试,就看了大佬的,发现都是二分啊,迭代啊之类的,然后我就找了一个代码量稍微少一点的。

class Solution:
    def searchRange(self, nums: List[int], target: int) -> List[int]:
        tl, tr = len(nums), -1
        # i是索引,j是与之元素
        for i,j in enumerate(nums):
            if j == target:
                tl = min(tl, i)
                tr = max(tr, i)
        if tl == len(nums):
            tl = -1
        return [tl, tr]

这个代码非常的简洁,他的思路很清晰,大家仔细看一看就可以看懂了,如果j一直不等于target,那么就证明nums里面没有target,这样子tl也就不会改变,返回[-1, -1],这个里面最重要的应该就是中间的tl = min(tl, i) 和 tr = max(tr, i) 这两句话了,因为 i 返回的是nums的索引嘛,然后利用min和max就可以找到最开始的位置与最后的位置了。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容