面试题 17.10. 主要元素投票算法 C++

Posted 2021-08-06 ai52learn

数组中占比超过一半的元素称之为主要元素。给你一个 整数 数组，找出其中的主要元素。若没有，返回 -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

class Solution {
public:
    int majorityElement(vector<int>& nums) {
        //投票算法，第一次遍历找到候选值，第二次遍历判断候选值是否满足条件
        int c=-1,cnt = 0;
        for(int i = 0;i<nums.size();i++)
        {
            if(cnt==0)
            {
                c = nums[i];
                cnt = 1;
                continue;
            }
            if(nums[i]!=c)
            {
                cnt--;
            }
            else cnt++;
        }
        cnt = 0;
        for(int i =0 ; i<nums.size();i++)
        {
            if(nums[i]==c) cnt++;
        }
        return cnt>nums.size()/2?c:-1;
    }
};