LeetCode34：在排序数组中查找元素的第一个和最后一个位置

最新推荐文章于 2024-11-16 15:18:00 发布

柠檬不甜会酸

最新推荐文章于 2024-11-16 15:18:00 发布

阅读量294

点赞数 1

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： leetcode 数据结构 python 算法

本文链接：https://blog.csdn.net/weixin_45666660/article/details/110422184

LeetCode 专栏收录该内容

82 篇文章

订阅专栏

博客围绕在升序整数数组中查找目标值的开始和结束位置展开。给出题目描述及示例，提出可设计时间复杂度为O(log n)的算法。思路是用for循环遍历，记录目标值首次和最后出现位置。最后给出相应Python代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、题目

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

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

进阶：

你可以设计并实现时间复杂度为 O(log n) 的算法解决此问题吗？

二、示例

示例 1：

输入：nums = [5,7,7,8,8,10], target = 8
输出：[3,4]

示例 2：

输入：nums = [5,7,7,8,8,10], target = 6
输出：[-1,-1]

示例 3：

输入：nums = [], target = 0
输出：[-1,-1]

提示：

0 <= nums.length <= 105
-109 <= nums[i] <= 109
nums 是一个非递减数组
-109 <= target <= 109

三、思路

一个for循环，当第一次遍历到 nums[i] == target 时，st = i，之后，每次遍历到nums[i] == target时，ed = i。

若nums数组中只有一个nums[i] == target，即st != -1，但ed == -1，此时令ed == st。

四、代码

class Solution:
    def searchRange(self, nums, target):
        """
        :param nums: List[int]
        :param target: int
        :return: List[int]
        """
        st, ed = -1, -1
        for i in range(len(nums)):
            if nums[i] == target:
                if st == -1: st = i
                else:ed = i
        if st != -1 and ed == -1:ed = st
        return [st, ed]

if __name__ == '__main__':
    nums = [5, 7, 7, 8, 8, 10]
    target = 8
    s = Solution()
    ans = s.searchRange(nums, target)
    print(ans)