leetcode 229. 求众数 II (摩尔投票进阶)

leetcode 229. 求众数 II

给定一个大小为 n 的整数数组，找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。

进阶：尝试设计时间复杂度为 O(n)、空间复杂度为 O(1)的算法解决此问题。

示例 1：

输入：[3,2,3]
输出：[3]

示例 2：

输入：nums = [1]
输出：[1]

示例 3：

输入：[1,1,1,3,3,2,2,2]
输出：[1,2]

提示：

• 1 <= nums.length <= 5 * 10^4
• -10^9 <= nums[i] <= 10^9

如果不了解摩尔投票算法，可以先看看这个 leetcode 面试题 17.10. 主要元素(摩尔投票算法)

只不过在原来的基础上，由一对数据对拼消耗改为两对数据对拼消耗。

class Solution {
public:
    vector<int> majorityElement(vector<int>& nums) {
        int n = nums.size();
        int ans1 = -1, ans2 = -1;
        int cnt1 =  0, cnt2 =  0;
        //需保证ans1 != ans2
        for(int x : nums){
            if(x == ans1) cnt1++;
            else if(x == ans2) cnt2++;
            else if(cnt1 == 0) {
                ans1 = x;
                cnt1 = 1;
            }
            else if(cnt2 == 0) {
                ans2 = x; 
                cnt2 = 1; 
            }//都不相同
            else{
                cnt1--;
                cnt2--;
            }
        }
        cnt1 = 0;
        cnt2 = 0;
        for(int x : nums){
            if(ans1 == x) cnt1++;
            if(ans2 == x) cnt2++;
        }
        
        vector<int> res;
        if(cnt1 > n / 3) res.push_back(ans1);
        if(cnt2 > n / 3 && ans2 != ans1) res.push_back(ans2);
        return res;
    }
};