给定一个大小为 n 的数组,找出其中所有出现超过 ⌊ n/3 ⌋
次的元素。
说明: 要求算法的时间复杂度为 O(n),空间复杂度为 O(1)。
示例 1:
输入: [3,2,3] 输出: [3]
示例 2:
输入: [1,1,1,3,3,2,2,2] 输出: [1,2]
Review:
认识到了一个新的算法---摩尔投票法
摩尔投票法的基本思想很简单,在每一轮投票过程中,从数组中找出一对不同的元素,将其从数组中删除。这样不断的删除直到无法再进行投票,如果数组为空,则没有任何元素出现的次数超过该数组长度的一半。如果只存在一种元素,那么这个元素则可能为目标元素。
通俗一点,如果一个数的个数超过了数组总数的一半,这个数就可以和别的数进行一换一,即删除自己一个数同时删除一个和自己不一样的数,那么最后剩下的一定是这个数,这个数处于'不败之地'...
同理,超过1/3也可以用相同的方式解决:
两个数A和B它们的个数都超过了总数的1/3,那么删除一个A和一个B,再删除一个与其他的数,到最后剩下的一定是A和B
这样就可以写出时间复杂度O(n),空间复杂度常数级的代码了:
Code
class Solution {
//摩尔投票法
public List<Integer> majorityElement(int[] nums) {
int c1 = 0, c2 = 0, n1 = 0, n2 = 0;
for (int num : nums) {
if (num == n1) {
c1++;
} else if (num == n2) {
c2++;
} else if (c1 == 0) {
n1 = num;
c1++;
} else if (c2 == 0) {
n2 = num;
c2++;
} else {
c1--;
c2--;
}
}
c1 = c2 = 0;
for (int num : nums) {
if (n1 == num) {
c1++;
} else if (n2 == num) {
c2++;
}
}
ArrayList<Integer> res = new ArrayList<>();
int p = nums.length / 3;
if (c1 > p) res.add(n1);
if (c2 > p) res.add(n2);
return res;
}
}
ps:我觉得把参数定义成 c1,c2....真的很骚