[Golang]力扣Leetcode—中级算法—其他—多数元素（哈希表|摩尔投票法）

[Golang]力扣Leetcode—中级算法—其他—多数元素（哈希表|摩尔投票法）

题目：给定一个大小为 n 的数组，找到其中的多数元素。多数元素是指在数组中出现次数 大于 ⌊ n/2 ⌋ 的元素。

你可以假设数组是非空的，并且给定的数组总是存在多数元素。

链接： 力扣Leetcode—中级算法—其他—多数元素.

示例 1：

输入：[3,2,3]
输出：3

示例 2：

输入：[2,2,1,1,1,2,2]
输出：2

标签：数组、哈希表、分治、计数、排序

思路：简单的想法是使用哈希表或者排序完成，遍历一次数组，将每个数出现的次数存入哈希表，当次数大于n/2时返回即可，代码如下：

全部Go代码如下：

package main
 
import (
	"fmt"
)
 
func majorityElement(nums []int) int {
	var res int
	m := make(map[int]int)
	n := len(nums)
 
	if n < 1 {
		return 0
	} else if n == 1 {
		return nums[0]
	} else {
		for _, v := range nums {
			m[v]++
		}
		for key, v := range m {
			if v > n/2 {
				res = key
				break
			} else {
				continue
			}
		}
		return res
	}
}
 
func main() {
	a := []int{8, 8, 7, 7, 7}
	fmt.Println(majorityElement(a))
}

提交截图：

然而该方案空间复杂度为O(n)，不符合要求O(1)

我们还可以用 摩尔投票的思想 ，对拼消耗，就可以满足空间复杂度为O(1)的要求。

• 遍历整个数组，从第一个数开始，count = 1，遇到相同的就加1，遇到不同的就减1
• 减到0，从下一个数开始计数。
• 由于多数元素的数量超出半数，最终剩下的一定就是多数元素

代码如下：

package main
 
import (
	"fmt"
)
 
func majorityElement(nums []int) int {
    count := 1
    major := nums[0]
 
    for i := 1; i < len(nums); i++ {
        if major == nums[i] {
            count++
        } else {
            count--
            if count == 0 {
                major = nums[i+1]
            }
        }
    }
    return major
}
 
func main() {
	a := []int{8, 8, 7, 7, 7}
	fmt.Println(majorityElement(a))
}