前言
求K众数,直观做法就是hash记录,然后取最大个数的那K个。而对于取众数,可用摩尔投票法将时间空间降低到O(n)/O(1).
当然,前提是你能分析出来是众数问题,有些问题很隐式。
一、多数元素II
二、直观hash|摩尔投票法
1、直观hash
// 每三个数都有它,说明这种数最多2个,最少0个。
// 暴力计算
public List<Integer> majorityElement(int[] nums) {
int n = nums.length;
int bound = n / 3;
Map<Integer, Integer> fx = new HashMap<>();
List<Integer> ans = new ArrayList<>();
for (int num : nums) fx.put(num, fx.getOrDefault(num, 0) + 1);
for (Map.Entry<Integer, Integer> entry : fx.entrySet()) if (entry.getValue() > bound) ans.add(entry.getKey());
return ans;
}
2、摩尔投票法
// 每三个数都有它,说明这种数最多2个,最少0个。
// 既然有0-2个,就先用投票法,选两个最多元素的出来,再次for循环记录实际有多少,最后确定两者是否大于 n/3,来确定是0/1/2个。
public List<Integer> majorityElement2(int[] nums) {
int el1 = 1 << 31, el2 = 1 << 31;//两个元素。
int v1 = 0, v2 = 0;//两者得票数
// 投票,确定最多的两个元素是谁?
for (int num : nums) {
// 注:这里一定要先判定是否有相等的了。
if (num == el1) ++v1;
else if (num == el2) ++v2;
else if (v1 == 0) {
el1 = num;
++v1;
} else if (v2 == 0) {
el2 = num;
++v2;
} else {
--v1;
--v2;
}
}
// 计算,确定两个最多元素,它实际的个数是多少。
v1 = v2 = 0;
for (int num : nums) {
if (num == el1) ++v1;
if (num == el2) ++v2;
}
// 判定,看最大元素是否满足条件。
int bound = nums.length / 3;
List<Integer> ans = new ArrayList<>();
if (v1 > bound) ans.add(el1);
if (v2 > bound) ans.add(el2);
// 返回,返回最最终满足条件的元素。
return ans;
}
总结
1)摩尔投票法解决一类众数问题。
参考文献
[1] LeetCode 多数元素II