题干:
给定一个大小为 n
的数组 nums
,返回其中的多数元素。多数元素是指在数组中出现次数 大于 ⌊ n/2 ⌋
的元素。
你可以假设数组是非空的,并且给定的数组总是存在多数元素。
解法:
func majorityElement(nums []int) int {
candidate, count := nums[0], 0
for _, num := range nums {
if count == 0 {
candidate = num
count = 1
} else if candidate == num {
count++
} else {
count--
}
}
return candidate
}
解析:
这道题是投票算法的直接应用。
- 遍历数组,维护一个候选元素
candidate
和一个计数器count
。 - 当计数器
count
为0时,设置当前元素为候选元素candidate
。 - 比较当前元素与
candidate
,如果相同则增加计数器count
,否则减少计数器count
。 - 最终遍历结束后的
candidate
即为所求的多数元素。
简单理解:
因为题目说给的数组一定存在超过一半的多数元素。所以不同元素的个数相互抵消,剩下的一定是多数元素。
具体示例:
假设有一个数组 nums = [3, 2, 3, 2, 3],我们需要找到其中的多数元素,即出现次数大于 ⌊ n/2 ⌋ 的元素。在这个例子中,n 是数组的长度,n = 5,所以多数元素需要出现至少 ⌊ 5/2 ⌋ = 3 次。
步骤说明:
初始化:
candidate:初始设为 nums[0],即 3。
count:初始设为 0。
遍历数组:
第一个元素 3:count 为0,设置 candidate 为 3,并将 count 设为1。
candidate = 3, count = 1
第二个元素 2:count 不为0,且 candidate 不等于 2,因此 count 减1。
candidate = 3, count = 0
第三个元素 3:count 为0,设置 candidate 为 3,并将 count 设为1。
candidate = 3, count = 1
第四个元素 2:count 不为0,且 candidate 不等于 2,因此 count 减1。
candidate = 3, count = 0
第五个元素 3:count 为0,设置 candidate 为 3,并将 count 设为1。
candidate = 3, count = 1
结果:
遍历结束后,candidate 为 3