LeetCode 229 Majority Element II

题目链接

题目大意：
给一个长度为\(n\)的数组，找出数组中出现次数大于\(\left \lfloor n/3 \right \rfloor\)的所有元素。

思路：
本题做法基于一个事实：每次选三个不同的数，从数组中删除，不断重复。该步骤最多只能执行\(\left \lfloor n/3 \right \rfloor\)遍，那么如果有一个元素出现大于\(\left \lfloor n/3 \right \rfloor\)次，那么它必然出现在剩下的数中。

代码(16ms)：

class Solution {
public:
    vector<int> majorityElement(vector<int>& nums) {
        if (nums.empty())
            return {};
        int x = nums.front(), cx = 0, y = nums.front(), cy = 0;
        for (int i : nums) {
            if (i == x)
                cx++;
            else if (i == y)
                cy++;
            else if (cx == 0)
                x = i, cx++;
            else if (cy == 0)
                y = i, cy++;
            else
                cx--, cy--;
        }
        
        vector<int> res;
        int k = nums.size() / 3;
        if (cx && count(nums.begin(), nums.end(), x) > k)
            res.push_back(x);
        if (cy && count(nums.begin(), nums.end(), y) > k)
            res.push_back(y);
        return res;
    }
};