做完LeetCode 169. Majority Element 后趁热打铁,做了Majority Element II。
这道题和169相似。差别在于,要找出所有出现次数大于n/3的数字,注意数字可能是不存在的。
做法也和169相似,注意到数字最多只有两个,所以这次需要假设两个Majority Element,而且要用到两个计数器。当出现三个不同的数字时,两个计数器都要减一。需要特别注意的是最后留下来的数字可能并不是Majority Element,所以要再遍历一次数组做一次确定。算法时间复杂度依然为O(N)。
代码如下:
# include <iostream>
# include <cstring>
# include <cstdio>
# include <vector>
using namespace std ;
class Solution {
public:
vector<int> majorityElement(vector<int>& nums) {
int me1, me2 ;
int ct1 = 0, ct2 = 0 ;
int l = nums.size() ;
vector<int> ans ;
for (int i=0; i<l; i++) {
if (!ct1 && (!ct2 || nums[i] != me2)) {
me1 = nums[i] ; ct1++ ;
}
else if (nums[i] == me1) ct1++ ;
else if (!ct2) {
me2 = nums[i] ; ct2++ ;
}
else {
if (nums[i] == me2) ct2++ ;
else {
ct1-- ; ct2-- ;
}
}
// printf("%d:%d %d:%d\n", me1, ct1, me2, ct2) ;
}
ct1 = ct2 = 0 ;
for (int i=0; i<l; i++) {
if (nums[i] == me1) ct1++ ;
if (nums[i] == me2) ct2++ ;
}
if (ct1 > l/3) ans.push_back(me1) ;
if (ct2 > l/3 && me2 != me1) ans.push_back(me2) ;
return ans ;
}
};