Leetcode--------数组（二分法）在排序数组中查找数字

原创已于 2024-09-12 13:10:38 修改 · 415 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #python

于 2020-07-16 23:31:12 首次发布

机器学习同时被 3 个专栏收录

27 篇文章

订阅专栏

leetcode

25 篇文章

订阅专栏

深度学习

23 篇文章

订阅专栏

本文介绍了一种高效算法，用于在已排序数组中查找特定目标值的出现次数。通过两次二分查找，分别定位目标值的左右边界，实现O(logN)的时间复杂度。适用于大数据量的搜索场景。

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

统计一个数字在排序数组中出现的次数（）。

示例 1:

输入: nums = [5,7,7,8,8,10], target = 8
输出: 2

示例 2:

输入: nums = [5,7,7,8,8,10], target = 6
输出: 0

限制：

0 <= 数组长度 <= 50000

思路一：

这个思路也是最容易想到的，从左边开始遍历找到第一个出现的 target ，位置用表示；从右开始遍历找到第一个出现的 target ，位置用表示，那么 target 在排序数组中出现的次数为 j-i+1 。这种方法的时间复杂度O(n)，空间复杂度O(1)。

代码：

class Solution(object):
    def search(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        i = 0
        j = len(nums)-1
        while i <= j and nums[i] != target:
            i+=1
        while j >= 0 and nums[j] != target :
            j-=1
        if i > j:
            return 0
        else:
            return j-i+1

思路二：

采用二分法的思想来做，下面我们给出二分法的步骤：

初始化：左边界，右边界。
循环二分：当 ,即闭区间无元素时跳出；
计算中点（向下取整）；
若，则在闭区间中，因此执行；
若，则在闭区间中，因此执行；
若，则返回m，退出循环。

思路：

排序数组中的所有数字形成一个窗口，记窗口的左 / 右边界索引分别为和，分别对应窗口左边 / 右边的首个元素。

本题要求统计数字的出现次数，可转化为：使用二分法分别找到左边界和右边界，易得数字的数量为。

具体步骤：

初始化：左边界，右边界。
循环二分：当 ,即闭区间无元素时跳出；
计算中点（向下取整）；
若，则在闭区间中，因此执行；
若，则在闭区间中，因此执行；
若，则右边界在闭区间中；左边界在闭区间中。因此分为以下两种情况：
（1）若查找右边界，则执行；（跳出时指向右边界）
（2）若查找左边界，则执行；（跳出时指向左边界）

复杂度分析：

时间复杂度 O(log N)：二分法为对数级别复杂度。
空间复杂度 O(1)：几个变量使用常数大小的额外空间。

代码：

class Solution:
    def search(self, nums: [int], target: int) -> int:
        # 搜索右边界 right
        i, j = 0, len(nums) - 1
        while i <= j:
            m = (i + j) // 2
            if nums[m] <= target: i = m + 1
            else: j = m - 1
        right = i
        # 若数组中无 target ，则提前返回
        if j >= 0 and nums[j] != target: return 0
        # 搜索左边界 left
        i = 0
        while i <= j:
            m = (i + j) // 2
            if nums[m] < target: i = m + 1
            else: j = m - 1
        left = j
        return right - left - 1

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zai-pai-xu-shu-zu-zhong-cha-zhao-shu-zi-lcof