算法中的数学问题 众数问题

169 多数元素

给定一个大小为 n 的数组,找到其中的多数元素。多数元素是指在数组中出现次数 大于 n / 2 的元素。

输入一个一维数组,输出一个整数,表示数组中个数占多数的元素。

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

解析:

​ 采用哈希表是一种最为直接的方法,统计数组中每个元素出现的次数,然后找出频次最高的元素即可。

class Solution {
public:
    int majorityElement(vector<int>& nums) {
        unordered_map<int,int> counts;
        for(const auto elem:nums){
            if(counts.find(elem) != counts.end()){
                counts[elem]++;
            }else{
                counts.insert(make_pair(elem,1));
            }
        }
        int ans = 0;
        int max = INT_MIN;
        for(const auto count: counts){
            if(count.second > max){
                ans = count.first;
                max = count.second;
            }
        }
        return ans;
    }
};

​ 本题也可以采用摩尔投票法更加简洁的解决。摩尔投票法的基本原理是:在数组中找到两个不相同的元素并删除它们,不断重复此过程,直到数组中元素都相同,那么剩下的元素就是主要元素。

​ 这种找主要元素的方法是怎么得以实现的呢?该投票法,遇到相同的元素则票数加 1,遇到不同的元素则票数减 1,当该元素的票数为 0 时更换投票主要元素。在主要元素个数大于 n / 2 的情况下,相当于每个主要元素和其他元素两两相互抵消,抵消到最后肯定还剩余至少 1 个多数元素。

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

229 求众数 II

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

输入一个一维数组,输出一个一维数组包含所有频次超过 n/3 的元素

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

解析:

​ 本题是摩尔投票法的进阶用法,摩尔投票法的核心是通过抵消计票选出多数,可以推出规律:如果至多选一个代表,那他的票数至少要超过一半 1/2 的票数;如果至多选两个代表,那他们的票数至少要超过 1/3 的票数;类推可知,如果至多选 m 个代表,那他们的票数至少要超过 1 / (m+1) 的票数。

​ 回到本题,求频次超过 n/3 的元素,那么最多只有两个,要注意多个主要元素时的抵消情况:(1)A同B不同,A++,B不变;(2)A不同B同,A不变,B++;(3)AB均不同,A–,B–;。

class Solution {
public:
    vector<int> majorityElement(vector<int>& nums) {
        vector<int> res;        
        const int len = nums.size();		
        int major1, major2;        
        int count1 = 0, count2 = 0;        
        for(int i = 0; i < len; i++) {
            // 存在元素相同
            if(nums[i] == major1){
                ++count1;
            }
            else if(nums[i] == major2){
                ++count2;
            }
            // 替换主要元素
            else if(count1 == 0) {
				major1 = nums[i];
				count1 = 1;
			}
            else if(count2 == 0) {
				major2 = nums[i];
				count2 = 1;
			}
            // 与major1和major2都不同时才抵消,如果只有一个不同的情况是另一个计数增加,不要想成了让不同的那一个去抵消,这里容易出错
            else {
				count1--;
				count2--;
			}
        }
        // 检验结果中元素数目是否超过三分之一
        count1 = 0, count2 = 0;
        for(auto &num : nums) {
            if(num == major1) count1++;
            else if(num == major2) count2++;
        }

        if(count1 > len / 3) res.push_back(major1);
        if(count2 > len / 3) res.push_back(major2);
        return res;
    }
};

参考资料

LeetCode 101:和你一起轻松刷题(C++) 第 9 章 巧解数学问题

上一篇:猜拳游戏——比积分,先的到3积分的赢


下一篇:leetcode-求众数 II