LeetCode 229. 求众数 II（Majority Element II ）

题目描述

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

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

示例 1:

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

示例 2:

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

解题思路

由于数组中出现次数超过 $frac{1}{3}$ 的数字最多只可能为两个，所以记录两个数字n1、n2，以及他们出现的次数c1、c2，遍历数组并做以下操作：

• 若当前两数字出现则把对应的次数加1；
• 若其中一个出现次数为0，则把当前数字赋给出现次数为0的数字，并将其出现次数置为1；
• 若当前数字不同于任何一个数字，则将两数字的出现次数都减1

最后得到两个数字以及他们出现的次数，再遍历一遍数组记录他们的出现次数，若大于 $frac{n}{3}$ 则加入到结果中。

代码

class Solution {
public:
    vector<int> majorityElement(vector<int>& nums) {
        vector<int> res;
        if(nums.empty()) return res;
        int n1 = nums[0], n2 = 0, c1 = 1, c2 = 0;
        for(int i = 1; i < nums.size(); i++){
            if(nums[i] == n1) c1++;
            else if(nums[i] == n2) c2++;
            else if(c1 == 0){
                n1 = nums[i];
                c1++;
            }
            else if(c2 == 0){
                n2 = nums[i];
                c2++;
            }
            else{
                c1--;
                c2--;
            }
        }
        c1 = c2 = 0;
        for(int i = 0; i < nums.size(); i++){
            if(nums[i] == n1) c1++;
            else if(nums[i] == n2) c2++;
        }
        if(c1 > nums.size() / 3) res.push_back(n1);
        if(c2 > nums.size() / 3) res.push_back(n2);
        return res;
    }
};