题目:给定一个大小为 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))
}

土豆
14 声望5 粉丝