给定一个大小为 n 的整数数组,找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。
进阶:尝试设计时间复杂度为 O(n)、空间复杂度为 O(1)的算法解决此问题。
示例 1:
输入:[3,2,3]
输出:[3]
示例 2:
输入:nums = [1]
输出:[1]
示例 3:
输入:[1,1,1,3,3,2,2,2]
输出:[1,2]
提示:
1 <= nums.length <= 5 * 104
-109 <= nums[i] <= 109
思路:摩尔投票
在任何数组中,出现次数大于该数组长度一半的值只能有一个。
在任何数组中,出现次数大于该数组长度1/3的值最多有只有两个。
由于最多只有两个元素,假设为A,B,则AB指的是出现次数最大的两个,不一定满足大于n/3,于是我们每次拿走三个不一样的数,最后剩下的一定是A,B
算法核心:每一次看三张牌,若三张票不同则抵消
三张不同的牌抵消可以分为三种情况:
1)包括A、B
把柱形图里的三个柱形都减一,得到的新柱形图最高的仍然是A、B,不影响最终结果
2)只包括A、B其中一个,设为A
虽然A减少了1,但是其它数的柱形也减少了1,得到的新柱形图最高的仍然是A、B,不影响最终结果
3)都不包括A、B
其它数的柱形减少,得到的新柱形图仍然是A、B最高,不影响最终结果
可以看到,三异牌对抗的目的是,每一次对抗都能使其它数的柱形减少,最后留下来的就是结果
于是可以有以下思路:
要求 > N / k为众数,则这样的数最多有k - 1个 因此维护两个候选变量cand1和cand2 两个阶段,抵消 + 计数 cand1和cand2的初始值任意,这里设为0,只要发现有计数为0,则马上更新新的cand 每考察到一个数,只有以下5种情况:
是cand1,则计数器++
是cand2,则计数器++
都不是,且两个计数器都 > 0,则两个计数器–(三个不同,抵消)
都不是,且count1是0,则让新的数成为cand1,且count1++(抵消至候选元素归零,则替换新的候选元素)
都不是,且count2是0,则让新的数成为cand2,且count2++(同上)
最后对cand1,cand2进行计数,若大于n/3,则放入答案。
ps:
本质上是三个一比。
当且仅当碰到的数既不是cand1也不是cand2,且两个计数器都 > 0时,计数器才会–,其他情况一律不减。
cand1和cand2可能一样,所以在计数阶段每个数只能算到一个候选人头上,即必须用if else语句。
AC代码:
class Solution {
public:
vector<int> majorityElement(vector<int>& nums) {
vector<int> ans;
if (nums.size() == 0) {
return ans;
}
int cand1 = 0, cand2 = 0, count1 = 0, count2 = 0, len = nums.size();
//抵消
for (int i = 0; i < len; i++) {
if (nums[i] == cand1) {
count1++;
} else if (nums[i] == cand2) {
count2++;
} else if (count1 > 0 && count2 > 0) {
count1--;
count2--;
} else if (count1 == 0) {
cand1 = nums[i];
count1++;
} else {
cand2 = nums[i];
count2++;
}
}
//计数
count1 = 0;
count2 = 0;
for (int i = 0; i < len; i++) {
if (nums[i] == cand1) {
count1++;
} else if (nums[i] == cand2) {
count2++;
}
}
if (count1 > len / 3) ans.push_back(cand1);
if (count2 > len / 3) ans.push_back(cand2);
return ans;
}
};