摩尔投票算法的改进的代码:
1,题目: LeetCode 229 [Majority Element II]
给定一个整型数组,找到所有主元素,它在数组中的出现次数严格大于数组元素个数的三分之一。
算法:每次删除三个不相同的数,最后留下的一定是出现次数超过1/3的数,这个思想可以推广到出现次数超过1/k次的元素有哪些。
因为出现次数大于n/3的元素最多只有两个,所以最开始可以维护两个数字(num1,num2)和两个计数器(counter1,counter2);
遍历数组,当数组中元素和num1或者num2相同,对应的counter1或者counter2加1;
如果counter1或counter2为0http://www.aivtp.com/将遍历到的该元素赋给num1或者nums2;
否则counter1和counter2都减1。
2,python代码:
num1,count1 = None,0
num2,count2 = None,0
for x in nums:# 算法核心,找出主要元素的候选值
if x == num1:
count1 += 1
elif x == num2:
count2 += 1
elif count1 == 0:
num1,count1 = x,1
elif count2 == 0:
num2,count2 = x,1
else:
count1 -= 1
count2 -= 1
count1,count2 = 0,0
for x in nums:# 统计确定候选值是真的主要元素
if x == num1:
count1 += 1
if x == num2:
count2 += 1
res = []
if count1 > len(nums)//3:
res.append(num1)
if count2 > len(nums)//3:
res.append(num2)
return res