【剑指Offer】37.数字在排序数组中出现的次数

题目描述

统计一个数字在排序数组中出现的次数。例如输入排序数组{1,2,3,3,3,3,4,5}和数字3，由于3在这个数组中出现了4次，因此输出4

解题思路

简单法

因为数组是排序的，自然而然的想到用二分查找法。先用二分查找算法找到一个3，然后在找到的3的左右两边顺序扫描，分别找出第一个3和最后一个3。因为要查找的数字在长度为n的数组中有可能出现O(n)次，所以顺序扫描的时间复杂度是O(n)。

思考如何更好地利用二分查找算法

有没有可能用二分查找算法直接找到第一个k及最后一个k呢？

那数组中间的数字和k作比较，如果中间的数字比k大，那么k只有可能出现在数组的前半段，下一轮只在数组的前半段查找就可以。后半段同理。

如果中间数字和k相等，如果前面一个数字也是k，那么第一个k肯定在数组的前半段，下一轮在数组前半段查找。

通过递归找到排序数组中的第一个k，同样的思路找到最后一个k：如果中间数字比k小，k只能出现在后半段，如果中间数字等于k，需要判断这个是不是最后一个k。然后最后一个k的位置-第一个k的位置+1就是个数

因为用的二分查找法，所以时间复杂度是O(logn)

Python代码

# -*- coding:utf-8 -*-
class Solution:
    def GetNumberOfK(self, data, k):
        # write code here
        num = 0
        if data and len(data) > 0:
            print("----")
            first = self.GetFirstK(data, k, 0, len(data)-1)
            last = self.GetLastK(data, k, 0, len(data)-1)
            print("1_first,end",first,last)
            if first > -1 and last > -1:
                print("first,end",first,last)
                num = last - first + 1
        return num
    
    def GetFirstK(self, data, k, start, end):
        if start > end:
            return -1
        middle_index = (start + end) // 2
        middle_data = data[middle_index]
        
        if( middle_data == k):
            if (middle_index > 0 and data[middle_index - 1] != k) or middle_index == 0:
                return middle_index
            else:
                end = middle_index - 1
        elif middle_data > k:
            end = middle_index - 1
        else:
            start = middle_index + 1
        return self.GetFirstK(data, k, start, end)
    
    def GetLastK(self, data, k, start, end):
        if start > end:
            return -1
        middle_index = (start + end) // 2
        middle_data = data[middle_index]
        
        if( middle_data == k):
            if (middle_index < len(data) - 1 and data[middle_index + 1] != k) or middle_index == len(data) -1:
                return middle_index
            else:
                start = middle_index + 1
        elif middle_data < k:
            start = middle_index + 1
        else:
            end = middle_index - 1
        return self.GetLastK(data, k, start, end)