Majority Element / Majority Element 2-主要元素

来源:互联网 发布:js 遇见未来下载 编辑:程序博客网 时间:2024/05/16 10:21

一类找主要元素的题目。

最简单最直观的想法就是,统计每个元素出现的个数(可以用map),然后再判断哪些元素是主要元素。时间和空间复杂度都是O(n)。

但是,Majority Element 2要求空间复杂度为O(1)。这就没想出来了。

网上搜了下,这是一类问题,即找主要元素的问题,采用moore投票法的思想

Majority Element:

摩尔投票法 Moore Voting,需要O(n)的时间和O(1)的空间,具体步骤见代码。

代码如下:

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


Majority Element 2:

解题思路:http://bookshadow.com/weblog/2015/06/29/leetcode-majority-element-ii/


题目中给了一条很重要的提示,让我们先考虑可能会有多少个众数,很明显,这种情况下,最多只有两个众数。即数组中至多可能会有2个出现次数超过 ⌊ n/3 ⌋ 的众数。


上一题是一个众数,这一题找两个众数,解法都一样。


记变量n1, n2为候选众数; c1, c2为它们对应的出现次数


遍历数组,记当前数字为num


若num与n1或n2相同,则将其对应的出现次数加1


否则,若c1或c2为0,则将其置为1,对应的候选众数置为num


否则,将c1与c2分别减1


最后,再统计一次候选众数在数组中出现的次数,若满足要求,则返回之。

代码:

class Solution {public:    vector<int> majorityElement(vector<int>& nums) {        int size = nums.size();        vector<int> res;        int th = size/3;        int c1 = 0, c2 = 1;        int n1 = 0, n2 = 0;        for(int i = 0; i < size; ++i){            if(nums[i]==c1)                n1++;            else if(nums[i]==c2)                n2++;            else if(n1==0){                n1 = 1;                c1 = nums[i];            }            else if(n2==0){                n2 = 1;                c2 = nums[i];            }            else{                n1--;                n2--;            }        }        n1=0,n2=0;        for(int i = 0; i < size; ++i){            if(nums[i]==c1) n1++;            if(nums[i]==c2) n2++;        }        if(n1>th) res.push_back(c1);        if(n2>th) res.push_back(c2);        return res;    }};



0 0
原创粉丝点击