53-1：数字在排序数组中出现的次数（剑指offer第2版Python）

1、题目描述

2、代码详解

二分查找的扩展。可以构造两个函数。

第一个函数查找目标数字出现的最前面的位置，先使用二分查找找到该数字，如果该数字的index > 0而且该数字前面一个数字等于k的话，那么就令end=middle-1，继续二分查找。

对于第二个函数，查找目标数字出现的最后面的位置，反之编写。最后如果数字存在的话，令走后面的index减去最前面的index然后+1即可。

在进行有序数组的元素查找，可以先尝试一下二分查找。

# -*- coding:utf-8 -*-
class Solution:
    def GetNumberOfK(self, data, k):
        count = 0
        if data != None and len(data) > 0:
            first = self.GetFirstK(data, k)
            last = self.GetLastK(data, k)
            if first > -1:
                # first和last只能同时== -1
                count = last - first + 1
        return count

    def GetFirstK(self, data, k):
        low, high = 0, len(data)-1
        while low <= high:
            mid = low + (high - low) // 2
            if data[mid] == k:
                if mid == 0 or data[mid-1] != k:  # 如果mid==0，则这个元素已经是数组第一个元素
                    # 如果mid != 0 ,则要看它前一个元素data[mid-1]
                    return mid  # 返回下标
                else:
                    high = mid - 1
            elif data[mid] > k:
                high = mid - 1
            else:
                low = mid + 1
        return -1
    def GetLastK(self, data, k):
        low, high = 0, len(data)-1
        while low <= high:
            mid = low + (high - low) // 2
            if data[mid] == k:
                if mid == len(data)-1 or data[mid+1] != k:
                    return mid  # 返回下标
                else:
                    low = mid + 1
            elif data[mid] > k:
                high = mid - 1
            else:
                low = mid + 1
        return -1
s = Solution()
data = [1, 2, 3, 3, 3, 3, 4, 5]
data2 = [1, 2, 3, 4]
data3 = [1, 2, 4]
print(s.GetNumberOfK(data, 3))
print(s.GetNumberOfK(data2, 2))
print(s.GetNumberOfK(data3, 3))