229. 求众数 II
给定一个大小为 n 的整数数组,找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。
示例 1:
输入:[3,2,3]
输出:[3]
示例 2:
输入:nums = [1]
输出:[1]
示例 3:
输入:[1,1,1,3,3,2,2,2]
输出:[1,2]
解题思路
摩尔投票法的变种,维护出现频次最大的两个元素,如果新元素不和这两个元素任意一个相等,则对这两个元素的出现频次进行抵消,一旦频次为0,最大的两个元素则被新加入的元素替换替换
-
假设只有一个元素出现超过 ⌊ n/3 ⌋ 次的元素,所以元素分为了2批,一批为出现超过 ⌊ n/3 ⌋ 次的元素n,另一批为除此以外的少于2/3元素a,在最极端的情况下,我们n中的元素不断被抵消,而a中的元素每次抵消也需要消耗两个元素,因为n的出现次数是大于1/3的,所以即使每次抵消3个元素以后,n最后仍然会剩余元素。
-
假设只有一个元素出现超过 ⌊ n/3 ⌋ 次的元素,所以元素分为了3批,两批为出现超过 ⌊ n/3 ⌋ 次的元素n1,n2,另一批为除此以外的少于1/3元素a,在最极端的情况下,最大元素为n1和a,因此新元素n2不断加入,来抵消n1和a的频次,但是因为n2的出现次数必定大于a,所以a最先会被抵消完,所以最大的元素就会被替换成为n1和n2
代码
func majorityElement(nums []int) []int {
n1,n2,cnt1,cnt2:=0,0,0,0
for _,i := range nums {
if cnt1>0&&i == n1 {
cnt1++
} else if cnt2>0&&i == n2 {
cnt2++
} else if cnt1 == 0 {
n1 = i
cnt1++
} else if cnt2 == 0 {
n2 = i
cnt2++
} else {
cnt1--
cnt2--
}
}
v1,v2:=0,0
for _,i := range nums {
if cnt1>0&&i==n1 {
v1++
}
if cnt2>0&&i==n2 {
v2++
}
}
res := []int{}
if cnt1>0&&v1>len(nums)/3 {
res = append(res, n1)
}
if cnt2>0&&v2>len(nums)/3 {
res = append(res, n2)
}
return res
}