Leetcode: Majority Element II

最新推荐文章于 2022-05-01 23:51:39 发布

Code Talk

最新推荐文章于 2022-05-01 23:51:39 发布

阅读量410

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/loushuai/article/details/50198163

算法专栏收录该内容

32 篇文章

订阅专栏

本文介绍了一种算法，用于在一个整数数组中找到所有出现次数超过数组大小1/3的元素，同时确保算法运行时间线性和空间复杂度为常数。

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.

这道题和找出现次数超过一半的数的意思类似，只不过这次最多只有两个元素可能满足条件。

class Solution {
public:
    vector<int> majorityElement(vector<int>& nums) {
        int cnt[2] = {0};
        int num[2] = {0};
        vector<int> ret;
        int i, j;
        const int n = nums.size();
        
        for (i = 0; i < n; ++i) {
            if (cnt[0] > 0 && num[0] == nums[i]) {
                cnt[0]++;
            } else if (cnt[1] > 0 && num[1] == nums[i]) {
                cnt[1]++;
            } else if (cnt[0] == 0) {
                cnt[0] = 1;
                num[0] = nums[i];
            } else if (cnt[1] == 0) {
                cnt[1] = 1;
                num[1] = nums[i];
            } else {
                cnt[0]--;
                cnt[1]--;
            }
        }
        if (cnt[0] > 0) {
            for (i = 0, j = 0; i < n; ++i) {
                if (nums[i] == num[0]) j++;
            }
            if (j > n/3) ret.push_back(num[0]);
        }
        if (cnt[1] > 0) {
            for (i = 0, j = 0; i < n; ++i) {
                if (nums[i] == num[1]) j++;
            }
            if (j > n/3) ret.push_back(num[1]);
        }
        
        return ret;
    }
};