给定一个大小为 n 的数组,找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。
说明: 要求算法的时间复杂度为 O(n),空间复杂度为 O(1)。
示例 1:
输入: [3,2,3]
输出: [3]
示例 2:
输入: [1,1,1,3,3,2,2,2]
输出: [1,2]
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/majority-element-ii 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
为了满足题目要求的复杂度,不能排序。 使用摩尔投票法
代码写的不优美,见谅。
class Solution {
public:
vector<int> majorityElement(vector<int>& nums) {
int n = nums.size(), cA = 0, cB = 0;
int numA, numB, i;
for(i = 0; i < n; i++)
{
if(cA == 0 || cB == 0)
{
if(cA == 0 && cB == 0)
{
numA = nums[i];
cA++;
}
else if(cA != 0 && cB == 0)
{
if(numA == nums[i])
cA++;
else
{
numB = nums[i];
cB++;
}
}
else //cA == 0 && cB != 0
{
if(numB == nums[i])
cB++;
else
{
numA = nums[i];
cA++;
}
}
}
else // 都不为零
{
if(nums[i] == numA)
cA++;
else if(nums[i] == numB)
cB++;
else
{
cB--;
cA--;
}
}
}
if(cA == 0 && cB == 0)
return {};
int nA=0, nB=0;
for(i = 0; i < n; i++)
{
if(nums[i] == numA)
nA++;
else if(nums[i] == numB)
nB++;
}
vector<int> ans;
if(nA >= n/3+1)
ans.push_back(numA);
if(nB >= n/3+1)
ans.push_back(numB);
return ans;
}
};