Leetcode 169. 229. 求众数 C++摩尔投票法

题目描述

Leetcode 169 求众数

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

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

示例 1:

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

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

思路

使用摩尔投票法，可以做到时间复杂读Ｏ(n)，空间复杂度Ｏ(1)。
使用摩尔投票的思路是：选择两个不同的元素，将这两个元素抛弃。那么最后留下的元素一定是众数。
参考：
https://mabusyao.iteye.com/blog/2223195
https://blog.csdn.net/weixin_42768679/article/details/81567231

解答

class Solution {
public:
    int majorityElement(vector<int>& nums) {
        int majority=nums[0];
        int count=1;
        for(int i=1;i<nums.size();++i)
        {
            if(0==count)
            {
                majority=nums[i];
                count++;
            }
            else
            {
                if(majority==nums[i]) count++;
                else count--;
            }
        }
        return majority;
    }
};

Leetcode 229. 求众数 II

class Solution {
public:
    vector<int> majorityElement(vector<int>& nums) {
        vector<int> res;
        int num1=0, num2=0;
        int count_1=0,count_2=0;
       
        for(int i=0;i<nums.size();++i)
        {
            if( (0==count_1 || nums[i]==num1) && nums[i]!=num2 )
            {
                num1=nums[i];
                count_1++;
            }
            else if(0==count_2 || nums[i]==num2)
            {
                num2=nums[i];
                count_2++;
            }
            else
            {
                count_1--;
                count_2--;
            }
        }
        count_1=0;
        count_2=0;
        for(int num : nums)
        {
            if(num==num1) count_1++;
            if(num==num2 && num1!=num2) count_2++;
        }
        if(count_1>nums.size()/3) res.push_back(num1);
        if(count_2>nums.size()/3) res.push_back(num2);
        return res;
    }
};