[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))
}