题目描述
Leetcode 169 求众数
给定一个大小为 n 的数组,找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。
说明: 要求算法的时间复杂度为 O(n),空间复杂度为 O(1)。
示例 1:
输入: [3,2,3]
输出: [3]
示例 2:
输入: [1,1,1,3,3,2,2,2]
输出: [1,2]
思路
使用摩尔投票法,可以做到时间复杂读O(n),空间复杂度O(1)。
使用摩尔投票的思路是:选择两个不同的元素,将这两个元素抛弃。那么最后留下的元素一定是众数。
参考:
https://mabusyao.iteye.com/blog/2223195
https://blog.csdn.net/weixin_42768679/article/details/81567231
解答
class Solution {
public:
int majorityElement(vector<int>& nums) {
int majority=nums[0];
int count=1;
for(int i=1;i<nums.size();++i)
{
if(0==count)
{
majority=nums[i];
count++;
}
else
{
if(majority==nums[i]) count++;//和前一个元素相同,统计这个元素还剩多少(注意是在不断丢弃元素)
else count--;//两个元素不同,同时丢弃
}
}
return majority;
}
};
Leetcode 229. 求众数 II
class Solution {
public:
vector<int> majorityElement(vector<int>& nums) {
vector<int> res;
int num1=0, num2=0;
int count_1=0,count_2=0;
for(int i=0;i<nums.size();++i)
{
if( (0==count_1 || nums[i]==num1) && nums[i]!=num2 )
{
num1=nums[i];
count_1++;
}
else if(0==count_2 || nums[i]==num2)
{
num2=nums[i];
count_2++;
}
else
{
count_1--;
count_2--;
}
}
count_1=0;
count_2=0;
for(int num : nums)
{
if(num==num1) count_1++;
if(num==num2 && num1!=num2) count_2++;
}
if(count_1>nums.size()/3) res.push_back(num1);
if(count_2>nums.size()/3) res.push_back(num2);
return res;
}
};