题目来源:https://leetcode-cn.com/problems/majority-element-ii/
大致题意:
统计一个数组出现次数大于 数组长度 / 3 的数
思路
- 哈希表统计次数,之后遍历哈希表,将次数大于 n/3 的取出
- 摩尔投票
摩尔投票
简单的说就是一个互相抵消的过程:
如果要求出现次数大于 n/3 的数,那么这样的数最多只有两个。假设两个数 a、b 的出现次数减去 n/3 分别为 x 和 y,那么可以将原数组分成三组:x 个 a,y 个 b,n - x - y 个剩余元素(由 a、b 和别的元素构成)。在最后一组中,又分别有 n-x-y / 3 个 a 和 b, n-x-y / 3 个其它元素,于是它们之间就可以相互抵消
遍历数组
- 若当前数等于 a 或者 b,那么其对应的计数 +1
- 若 a 或 b 对应的计数为 0,那么代表之前出现的 a 或 b 已经被抵消完,这时候将当前元素重置为 a 或 b
- 不满足前两种情况,也就代表当前数属于已有 a、 b 之外的其他元素,抵消一次,a 和 b 的计数 -1
代码:
public List<Integer> majorityElement_mole(int[] nums) {
int count1 = 0;
int count2 = 0;
int num1 = 0;
int num2 = 0;
for (int num : nums) {
if (count1 > 0 && num1 == num) { // 第一个数计数++
count1++;
} else if (count2 > 0 && num2 == num) { // 第二个数计数++
count2++;
} else if (count1 == 0) { // 重置第一个数
count1++;
num1 = num;
} else if (count2 == 0) { // 重置第二个数
count2++;
num2 = num;
} else { // 抵消
count1--;
count2--;
}
}
int countSum1 = 0;
int countSum2 = 0;
// 统计两个数(如果有的话)出现次数
for (int num : nums) {
if (count1 > 0 && num1 == num) {
countSum1++;
} else if (count2 > 0 && num2 == num) {
countSum2++;
}
}
int n = nums.length / 3;
List<Integer> ans = new ArrayList<>();
// 确保两个数出现次数大于 n/3
if (countSum1 > n) {
ans.add(num1);
}
if (countSum2 > n) {
ans.add(num2);
}
return ans;
}