【Leetcode】229. 求众数 II

题目描述：

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

说明: 要求算法的时间复杂度为 O(n)，空间复杂度为 O(1)。

示例 1:

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

示例 2:

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

解题思路：

本题根据时间和空间复杂度的要求，使用摩尔投票算法。

参考：https://www.hijerry.cn/p/45987.html

AC代码：

class Solution {
public:
    vector<int> majorityElement(vector<int>& nums) 
    {
        int len = nums.size();
        vector<int> res;

        if (len == 0)
            return res;
        if (len == 1)
        {
            res.push_back(nums[0]);
            return res;
        }
        //最多两个人可以大于n/3
        int c1 = nums[0];
        int c2 = nums[0];
        int cnt1 = 0;
        int cnt2 = 0;
        for (int i = 0; i<len; i++)
        {
            int cur = nums[i];
            if (cur == c1)
            {
                cnt1++;
            }
            else if (cur == c2)
            {
                cnt2++;
            }
            else
            {
                if (cnt1 == 0)
                {
                    c1 = cur;
                    cnt1 = 1;
                }
                else if (cnt2 == 0)
                {
                    c2 = cur;
                    cnt2 = 1;
                }
                else 
                { 
                    cnt1--;
                    cnt2--;
                }

            }
        }
        // 统计大多数的数出现次数
        cnt1 = 0;
        cnt2 = 0;
        for (int i = 0; i<len; i++)
        {
            if (c1 == nums[i])
                cnt1++;
            if (c2 == nums[i])
                cnt2++;
        }
        //判断是否>n/3
        if (cnt1 > len / 3)
        {
            res.push_back(c1);
        }
        if (c2 != c1 && cnt2 > len / 3)
        {
            res.push_back(c2);
        }

        return res;
    }
};

备注：

感谢Hugh_Ni 同学的指正，之前的解法时间复杂度超过线性，只是我AC过了就没有注意。自己百度了一下摩尔投票算法，重新解题。