剑指offer(15)数字在排序数组中出现的次数——python

题目描述
统计一个数字在排序数组中出现的次数。
例如：统计数组{1,2,3,3,3,3,4,5}中数字3出现的次数

思路：
利用二分查找法，二分查找法总是先拿数组中间的数和k作比较。如果中间的数字比k大，那么k只有可能出现在数组的前半段，下一轮只在数组的前半段查找即可。如果中间的数字比k小，那么k只有可能出现在数组的后半段，下一轮我们只在数组的后半段查找即可。如果中间数字和k相等，则看其前面一位的数字，如果中间的数字的前面一个数字不是k，则此时中间的数字刚好就是第一个k，如果中间数字的前面一个数字也是k，那么第一个k肯定在数组的前半段，下一轮仍需要在数组的前半段查找。
由上述过程可写出两个递归函数，分别求出所要查找的数字出现的第一位的索引和最后一位的索引
在函数GetFirstK中，如果数组中不包含数字k，则返回-1。如果数组中包含至少一个k，则返回第一个k在数组中的下标。同理，GetLastK函数也是，数组中不包含k则返回-1，数组中包含k就返回最后一个k在数组中的下标。

# -*- coding:utf-8 -*-
class Solution:
    def GetNumberOfK(self, data, k):
        # write code here\
        length = len(data)
        number = 0
        if data and length > 0:
            first = self.GetFirstK(data, length, k, 0, length - 1)
            last = self.GetLastK(data, length, k, 0, length - 1)
            print(first)
            print(last)
            if first > -1 and last > -1:
                number = last - first + 1
        return number

    def GetFirstK(self, data, length, k, start, end):
        if start > end:
            return -1
        mid = int((start + end) / 2)
        mid_data = data[mid]
        if mid_data == k:
            if mid > 0 and data[mid - 1] != k or mid == 0:
                return mid
            else:
                end = mid - 1
        elif mid_data > k:
            end = mid - 1
        else:
            start = mid + 1
        return self.GetFirstK(data, length, k, start, end)

    def GetLastK(self, data, length, k, start, end):
        if start > end:
            return -1
        mid = int((start + end) / 2)
        mid_data = data[mid]
        if mid_data == k:
            if mid < length - 1 and data[mid + 1] != k or mid == length - 1:
                return mid
            else:
                start = mid + 1
        elif mid_data < k:
            start = mid + 1
        else:
            end = mid - 1
        return self.GetLastK(data, length, k, start, end)