【题目】
给定一个整型数组arr,打印其中出现次数大于一半的数,如果没有这样的数,打印提示信息。要求时间复杂度为O(N),空间复杂度为O(1)。
【进阶】
给定一个整型数组arr,再给定一个整型K,打印所有出现次数大于N/K的数,如果没有这样的数,打印提示信息。要求时间复杂度为O(N*K),额外空间复杂度O(K)。
【基本思路】
对于原问题和进阶问题,都可以使用哈希表记录每个数出现的次数,但是额外空间复杂度为O(N),不符合题意。正确的做法是,一次在数组中删除k个不同的数,不停的删除,直到剩下的元素的种类不足k。如果一个数出现的次数大于N/K,那么这个数一定会被留下来。需要注意的是:出现N/K次的数一定会留下来,但是留下来的数未必就是出现N//K次的数,比如arr = [1, 2, 3],k = 2。
接下来要解决的问题就是如何不停删除k个不同的数,这里以 k = 2(出现次数大于一半)为例。
采用阵地攻守的思想:
第一个数字作为第一个士兵,守阵地;count = 1;
遇到相同元素,count++;
遇到不相同元素,即为敌人,同归于尽,count- -;当遇到count为0的情况,又以新的i值作为守阵地的士兵,继续下去,到最后还留在阵地上的士兵,有可能是主元素。
再加一次循环,记录这个士兵的个数看是否大于数组一般即可。
(摘自牛客网某网友分享)
具体实现如下:
首先定义两个变量,cand = 0,times = 0。cand表示目前的候选元素,times表示该元素出现的次数。依次遍历数组,如果当前times = 0表示此时没有候选元素,将arr[i]赋值给cand;如果times != 0,考虑两种情况:
<1> 如果arr[i] == cand,说明当前没有找到与cand不同的数,此时令cand出现的次数加1,即times+1;
<2> 如果 arr[i] != cand,说明此时找到了另一个不同的数,那么令cand的次数减1,如果减完之后times == 0,说明候选元素被消耗完了,数组中的下一个元素就又会变成候选元素。
如此反复,遍历完数组后,出现次数大于一半的数一定就是cand。这时我们还要再遍历一次数组,判断cand出现的次数是否大于一半。
下面是使用python3.5实现的代码:
def printHalfMajor(arr):
if arr == None or len(arr) == 0:
print("No such number!")
return
cand = 0
times = 0
for i in range(len(arr)):
if times == 0:
cand = arr[i]
times += 1
elif arr[i] == cand:
times += 1
else:
times -= 1
times = 0
for i in range(len(arr)):
if arr[i] == cand:
times += 1
if times > len(arr) // 2:
print(cand)
else:
print("No such number!")
进阶问题和原问题的解题思路一样,一次删除k个不同的数,直到剩下的数的种类不足k。原问题中我们,我们使用了一个候选cand找到了出现次数大于N / K的数,进阶问题中,我们只要使用 K - 1个候选元素即可。详情见如下代码:
def printKMajor(arr, k):
if arr == None or len(arr) == 0 or k < 2:
print("No such number!")
return
map = {}
moveMap = {}
for i in range(len(arr)):
if arr[i] in map:
map[arr[i]] += 1
else:
if len(map) == k-1:
for key in map:
map[key] -= 1
if map[key] == 0:
moveMap[key] = 1 #记录要删除的字典元素,不可直接删除
for key in moveMap:
del map[key]
moveMap.clear()
else:
map[arr[i]] = 1
times = 0
flag = True
for key in map:
for i in range(len(arr)):
if arr[i] == key:
times += 1
if times > len(arr) // k:
print(key,end=' ')
flag = False
times = 0
print("No such number!" if flag else '')