且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

[LeetCode] Majority Element II

更新时间:2022-01-27 23:37:37

Given an integer array of size n, find all elements that appear more than ⌊ n/3 ⌋ times. The algorithm should run in linear time and in O(1) space.

解题思路

摩尔投票法。投票法的核心是找出两个候选众数进行投票,需要两遍遍历,第一遍历找出两个候选众数,第二遍遍历重新投票验证这两个候选众数是否为众数即可。

实现代码

C++:

class Solution {
public:
    vector<int> majorityElement(vector<int>& nums) {
        vector<int> res;
        int m = 0, n = 0, cm = 0, cn = 0;
        for (int i = 0; i < nums.size(); i++)
        {
            if (nums[i] == m)
            {
                cm++;
            }
            else if (nums[i] == n)
            {
                cn++;
            }
            else if (cm == 0)
            {
                m = nums[i];
                cm = 1;
            }
            else if (cn == 0)
            {
                n = nums[i];
                cn = 1;
            }
            else
            {
                --cm;
                --cn;
            }
        }

        cm = cn = 0;
        for (auto& a : nums)
        {
            if (a == m)
            {
                cm++;
            }
            else if (a== n)
            {
                cn++;
            }
        }
        if (cm > nums.size() / 3)
        {
            res.push_back(m);
        }
        if (cn > nums.size() / 3)
        {
            res.push_back(n);
        }

        return res;
    }
};

Java:

public class Solution {
    public List<Integer> majorityElement(int[] nums) {
        List<Integer> res = new ArrayList<Integer>();
        int m = 0, n = 0, cm = 0, cn = 0;
        for (int a : nums) {
            if (a == m) {
                ++cm;
            } else if (a == n) {
                ++cn;
            } else if (cm == 0) {
                m = a;
                cm = 1;
            } else if (cn == 0) {
                n = a;
                cn = 1;
            } else {
                --cm;
                --cn;
            }
        }

        cm = cn = 0;
        for (int a : nums){
            if (a == m) {
                ++cm;
            } else if (a == n) {
                ++cn;
            }
        }

        if (cm > nums.length / 3) {
            res.add(m);
        }
        if (cn > nums.length / 3) {
            res.add(n);
        }

        return res;
    }
}