剑指offer：数字在排序数组中出现的次数

最新推荐文章于 2019-04-20 21:27:08 发布

IT莫莫

最新推荐文章于 2019-04-20 21:27:08 发布

阅读量232

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指offer

本文链接：https://blog.youkuaiyun.com/gt362ll/article/details/82751355

剑指offer 专栏收录该内容

49 篇文章

订阅专栏

本文介绍了一种使用二分查找算法确定已排序数组中特定元素出现次数的方法。通过两次二分查找分别定位该元素的起始和结束位置，从而高效地计算出元素的总数量。

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

# -*- coding:utf-8 -*-
class Solution:
    def GetNumberOfK(self, data, k):
        # write code here
        low=0
        high=len(data)-1
        mid=(low+high)/2
        while low<=high:
            mid=(low+high)/2
            if data[mid]<k:
                low=mid+1
            else:
                high=mid-1
            #mid=(low+high)/2
        start=low
        low=0
        high=len(data)-1
        mid=(low+high)/2
        while low<=high:
            if data[mid]<=k:
                low=mid+1
            else:
                high=mid-1
            mid=(low+high)/2
        end=high
        
        return end-start+1

题目给出的是一个排序好的数组，k一定是连续的，考虑用二分搜索来找到k有多少。即找到k的开始位置和结束位置。注意二分查找的一些点，例如where判断条件是<=，low和high更新的时候要+1 -1来保证每次范围都在缩小。然后因为要找到开始和结束的位置，当data[mid]==k的时候不能返回，而要继续探索，当要找左边界的时候要接着认为想找的k在左边，右边同理。所以才有第一个循环中if data[mid]>=k: high=mid-1