给定一个大小为 n 的数组,找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。 你的算法应该在O(1)空间中以线性时间运行。
超过n/3的元素最多只有两个,因此用num1,num2记录这两个数,count1,count2记录出现次数,采用抵消法,当再次遍历到nums[i]==num1 || nums[i]==num2时,count1++或者count2++,否则count1–,count2–,最后得到的num1,num2就是候选元素,最后还需要再遍历一遍,验证num1和num2是否超过n/3
public List<Integer> majorityElement(int[] nums) {
List<Integer> list = new ArrayList<Integer>();
if (nums.length == 1) {
list.add(nums[0]);
return list;
}
else if(nums.length == 2){
list.add(nums[0]);
if(nums[0] != nums[1]){
list.add(nums[1]);
}
return list;
}
int num1 = 0;
int num2 = 0;
int count1 = 0;
int count2 = 0;
for (int i = 0; i < nums.length; i++) {
if (count1 != 0 && nums[i] == num1) {
count1++;
}
else if(count2 != 0 && nums[i] == num2){
count2++;
}
else if(count1 == 0){
num1 = nums[i];
count1 = 1;
}
else if(count2 == 0){
num2 = nums[i];
count2 = 1;
}else {
count1 = count1 > 0 ? count1 - 1 : 0;
count2 = count2 > 0 ? count2 - 1 : 0;
}
}
count1 = 0;
count2 = 0;
for (int i = 0; i < nums.length; i++) {
if (nums[i] == num1) {
count1++;
} else if (nums[i] == num2) {
count2++;
}
}
if (count1 > nums.length / 3) {
list.add(num1);
}
if (count2 > nums.length / 3) {
list.add(num2);
}
return list;
}