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

Posted 2021-09-13 炫云云

剑指 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

二分查找

考虑 $\text{target}$ 开始和结束位置，其实我们要找的就是数组中「第一个等于 $\text{target}$ 的位置」（记为 $\text{leftIdx}$ ）和「第一个大于 $\text{target}$ 的位置减一」（记为 $\text{rightIdx}$）。

定义一个二分查找，寻找 $\text{leftIdx}$ 即为在数组中寻找第一个等于 $\text{target}$ 的下标，寻找 $\text{rightIdx}$ 即为在数组中寻找第一个大于 $\text{target}$ 的下标，然后将下标减一,即为最后一个位置。

最后，因为 $\text{target}$ 可能不存在数组中，因此我们需要重新校验我们得到的两个下标 $\text{leftIdx}$ 和 $\text{rightIdx}$，看是否符合条件，如果符合条件就返回 $\text{rightIdx}-\text{leftIdx}+1$ ，不符合就返回 $0$ 。

class Solution(object):
    def search(self, nums, target):
        """
        :type nums: List[int]
        :type target: int
        :rtype: int
        """
        def binarySearch(nums, target):
            left =0
            right = len(nums) -1
            while left<=right:
                mid = left+(right-left)//2
                if nums[mid]<target:# 应该继续向右边找,搜索区间变为 [mid+1, right]
                    left = mid+1
                else:
                    right = mid  -1  #应该继续向左边找 ,搜索区间变为 [left, mid  -1]
                    #寻找第一个满足条件的点， 即使得right在满足target的数的最左边,即寻找左边界
            return left
        leftIdx = binarySearch(nums, target)
        print(leftIdx)
        rightIdx = binarySearch(nums, target+1) -1
        print(rightIdx)

        if leftIdx == len(nums) or nums[leftIdx] !=target:
            return 0
        else:
            return rightIdx - leftIdx +1

s= Solution()
s.search(nums = [1], target = 1)

1
0
0

参考

Krahets - 力扣（LeetCode） (leetcode-cn.com)