No.243 - LeetCode[229] Majority Element II - 数组中找出重复多于N/3的数字

最新推荐文章于 2024-10-06 13:05:10 发布

ShellDawn

最新推荐文章于 2024-10-06 13:05:10 发布

阅读量145

点赞数

分类专栏： leetcode 搜索算法文章标签： leetcode

本文链接：https://blog.csdn.net/ShellDawn/article/details/109036274

版权

leetcode 同时被 2 个专栏收录

251 篇文章 0 订阅

订阅专栏

搜索算法

4 篇文章 0 订阅

订阅专栏

难点在于使用O(1)空间，O(N)时间

因为重复多于N/3，所以至多只有两个数

/*
 * @lc app=leetcode id=229 lang=cpp
 *
 * [229] Majority Element II
 */

// @lc code=start
class Solution {
public:
    vector<int> majorityElement(vector<int>& nums) {
        int A = 0,cntA = 0;
        int B = 0,cntB = 0;
        int N = nums.size();
        for(int i=0;i<N;i++){
            if(nums[i] == A) cntA++;
            else if(nums[i] == B) cntB++;
            else if(cntA == 0){
                A = nums[i];
                cntA = 1;
            }else if(cntB == 0){
                B = nums[i];
                cntB = 1;
            }else{
            	// 这里很精髓，如果有两个>n/3的数，则剩余数一定不能消耗完A和B
            	// 如果只有一个>n/3的数，B则会消耗掉(N-A)/2的数，剩余数一定比A少
                cntA -- ;
                cntB -- ;
            }
        }
        cntA = cntB = 0;
        for(int i=0;i<N;i++){
            if(nums[i] == A) cntA++;
            if(nums[i] == B) cntB++;
        }
        vector<int> ans;
        if(cntA > N/3) ans.push_back(A);
        if(cntB > N/3 && B != A) ans.push_back(B);
        return ans;
    }
};
// @lc code=end