题目描述
给定一个大小为 n
的数组 nums
,返回其中的多数元素。多数元素是指在数组中出现次数 大于 ⌊ n/2 ⌋
的元素。
你可以假设数组是非空的,并且给定的数组总是存在多数元素。
示例 1:
输入: nums = [3,2,3]
输出: 3
示例 2:
输入: nums = [2,2,1,1,1,2,2]
输出: 2
提示:
- n == nums.length
- 1 <= n <= 5 * 104
- -109 <= nums[i] <= 109
代码及注释
func majorityElement(nums []int) int {
res := 0
count := 0
for _, num := range nums {
if num == res {
count++ // 如果当前数字与候选数字相同,则计数加1
} else if count == 0 {
res = num // 如果当前数字与候选数字不同且计数为0,则更新候选数字
} else {
count-- // 如果当前数字与候选数字不同且计数不为0,则计数减1
}
}
return res
}
代码解释
使用的是摩尔投票算法来找出数组中的主要元素。摩尔投票算法是一种高效的算法,可以在O(n)的时间复杂度内找出数组中出现次数超过一半的元素(如果存在的话)。
这里的逻辑是:
- 使用变量
res
初始化为0(这里假设数组中没有0)和计数count
初始化为0。 - 遍历数组
nums
,对于数组中的每一个数字:- 如果当前数字与候选数字
res
相同,则计数count
加1。 - 如果当前数字与候选数字
res
不同且计数count
为0,则更新候选数字res
为当前数字。 - 如果当前数字与候选数字
res
不同且计数count
不为0,则计数count
减1。
- 如果当前数字与候选数字
- 最后,
res
就是数组中的主要元素。
这个方法的时间复杂度是O(n),其中n是数组nums
的长度。