LeetCode求众数：摩尔投票法golang实现

给定一个大小为 n 的整数数组，找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。

进阶：尝试设计时间复杂度为 O(n)、空间复杂度为 O(1)的算法解决此问题。

LeetCode原题：

作者：wotxdx
链接：https://leetcode-cn.com/problems/majority-element-ii/solution/liang-fu-dong-hua-yan-shi-mo-er-tou-piao-fa-zui-zh/

摩尔投票法，解决的问题是如何在任意多的候选人中，选出票数超过一半的那个人。注意，是超出一半票数的那个人。

摩尔投票法分为两个阶段：

                       抵消阶段和计数阶段。

抵消阶段：两个不同投票进行对坑，并且同时抵消掉各一张票，如果两个投票相同，则累加可抵消的次数；

计数阶段：在抵消阶段最后得到的抵消计数只要不为 0，那这个候选人是有可能超过一半的票数的，为了验证，则需要遍历一次，统计票数，才可确定。

摩尔投票法经历两个阶段最多消耗 O(2n) 的时间，也属于 O(n)的线性时间复杂度，另外空间复杂度也达到常量级。

如果最后得到的可抵消票数是 0 的话，那他已经无缘票数能超过一半的那个人了。因为本来可能有希望的，但是被后面的一张不同的票抵消掉了。所以，在这里可以直接返回结果，无需后面的计算了。

如果最后得到的抵消票数不为 0 的话，那说明他可能希望的.

理解摩尔投票法之后，我们再回到题目描述，题目可以看作是：在任意多的候选人中，选出票数超过⌊ 1/3 ⌋的候选人。

显然，票数超过1/3的最多只可以有2个候选人：

如果至多选一个代表，那他的票数至少要超过一半（⌊ 1/2 ⌋）的票数；

如果至多选两个代表，那他们的票数至少要超过 ⌊ 1/3 ⌋ 的票数；

如果至多选m个代表，那他们的票数至少要超过 ⌊ 1/(m+1) ⌋ 的票数。

所以以后碰到这样的问题，而且要求达到线性的时间复杂度以及常量级的空间复杂度，直接套上摩尔投票法。

我这里只是把原作者的代码重构了一下，把投票和计数部分单独封装了两个函数：

func majorityElement(nums []int) []int {
	// 创建返回值
	result := make([]int, 0)
	if nums == nil || len(nums) == 0 {
		return result
	}
	//投票和计数两个阶段
	cand1, cand2 := voteStage(nums)
	return countStage(cand1, cand2, nums)
}

投票函数：

/*投票阶段

初始化两个候选人 candidate，以及他们的计数票*/

func voteStage(nums []int) (cand1, cand2 int) {
	cand1, cand2 = nums[0], nums[0]
	count1, count2 := 0, 0
	//摩尔投票法
	for _, num := range nums {
		// 投票
		if cand1 == num {
			count1++
			continue
		}
		if cand2 == num {
			count2++
			continue
		}
		if count1 == 0 {
			cand1 = num
			count1++
			continue
		}
		if count2 == 0 {
			cand2 = num
			count2++
			continue
		}
		count1--
		count2--
	}
	return cand1, cand2
}

计数阶段：统计两个候选人出现的次数，并判断出现的次数是否超过总数的1/3.

func countStage(cand1, cand2 int, nums []int) (result []int) {
	count1, count2 := 0, 0
	for _, num := range nums {
		if cand1 == num {
			count1++
		} else if cand2 == num {
			count2++
		}
	}
	//判断候选人的出现频率是否大于1/3
	if count1 > len(nums)/3 {
		result = append(result, cand1)
	}
	if count2 > len(nums)/3 {
		result = append(result, cand2)
	}
	return result
}

测试：

import (
    "fmt"
)

func main(){
    candidate := []int{7, 2, 7, 2, 7, 5, 7, 2, 2, 2, 9}
	fmt.Println(majorityElement(candidate))

}

输出：

>>[7 2]