题目描述
统计一个数字在排序数组中出现的次数。例如输入排序数组{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)