剑指 Offer 53 - I. 在排序数组中查找数字 I

最新推荐文章于 2022-03-30 01:08:54 发布

diluosixu

最新推荐文章于 2022-03-30 01:08:54 发布

阅读量94

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/u012763126/article/details/115444841

版权

leetcode 专栏收录该内容

34 篇文章 1 订阅

订阅专栏

剑指 Offer 53 - I. 在排序数组中查找数字 I
统计一个数字在排序数组中出现的次数。

示例 1:

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

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

限制：

0 <= 数组长度 <= 50000

注意：本题与主站 34 题相同（仅返回值不同）：https://leetcode-cn.com/problems/find-first-and-last-position-of-element-in-sorted-array/

方法1：遍历计数

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        count = 0
        for num in nums:
            if num == target:
                count += 1
        return count

方法2：list.count()

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        return nums.count(target)

方法3：二分查找
算法解析：
(1) 初始化：左边界 i=0 ，右边界 j=len(nums)−1 。
(2) 循环二分：当闭区间 [i,j] 无元素时跳出；
计算中点 mid=(i+j)/2 （向下取整）；

若 nums[mid]<target ，则 target 在闭区间 [mid+1,j] 中，因此执行 i=mid+1；
若 nums[mid]>target ，则 target 在闭区间 [i,mid−1] 中，因此执行 j=mid−1；
若 nums[mid]=target ，则右边界 right 在闭区间 [mid+1,j] 中；左边界 left 在闭区间 [i, mid-1] 中。因此分为以下两种情况：
1)若查找右边界 right ，则执行i=mid+1 ；（跳出时 i 指向右边界）
2)若查找左边界left ，则执行 j=mid−1 ；（跳出时 j 指向左边界）
(3) 返回值：应用两次二分，分别查找right 和 left ，最终返回 right−left−1 即可。
效率优化：
找到target的右边界helper(target) ，target-1的右边界helper(target-1)，返回helper(target) - helper(target-1)即可。

class Solution:
    def search(self, nums: List[int], target: int) -> int:
        def helper(tar):
            i, j = 0, len(nums) - 1
            while i <= j:
                mid = (i + j) // 2
                if nums[mid] <= tar:
                    i = mid + 1
                else:
                    j = mid - 1
            return i

        return helper(target) - helper(target-1)