leetcode 面试题 17.10. 主要元素
数组中占比超过一半的元素称之为主要元素。给你一个 整数 数组,找出其中的主要元素。若没有,返回 -1 。请设计时间复杂度为 O(N) 、空间复杂度为 O(1) 的解决方案。
示例 1:
输入:[1,2,5,9,5,9,5,5,5]
输出:5
示例 2:
输入:[3,2]
输出:-1
示例 3:
输入:[2,2,1,1,1,2,2]
输出:2
摩尔投票算法:核心就是对拼消耗。
算法步骤:
- 初始化cnt = 0,用来计数;ans表示当前主要元素
- 遍历数组,如果接下来出现的数字与ans相同,cnt加1。如果不同,cnt减1。
- 如果cnt== 0,表示之前出现的所有数字都可以凑成不同的数对,相互抵消。大于1/2 n的数还会在后面出现。换个数重新开始计数。
- 由于不一定存在主要元素,因此需要第二次遍历数组进行验证
class Solution {
public:
int majorityElement(vector<int>& nums) {
int n = nums.size();
int ans = -1;
int cnt = 0;
for(int i=0; i<n; i++){
if(cnt == 0){
ans = nums[i];
cnt = 1;
}
else{
if(ans == nums[i]) cnt++;
else cnt--;
}
}
//验证ans是否为主要元素
cnt = 0;
for(int i=0; i<n; i++){
if(ans == nums[i]){
cnt++;
}
}
return cnt * 2 > n ? ans : -1;
}
};