leetcode 面试题 17.10. 主要元素(摩尔投票算法)

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;
    }
};