主元素问题

问题描述

主元素的定义为：数组中出现次数超过数组长度一半以上的元素。

输入一个无序数组，输出主元素（不能保证一定存在主元素）。

解决思路

经典的芯片测试问题：

1. 首先将数组的首元素置为主元素候选，并附加一个计数器，初始为1；

2. 遍历数组之后的元素，如果元素与候选元素相等，计数器加1；否则计数器减1；途中如果计数器为0，则将当前遍历元素设为候选元素。

因为不能保证最后得到的候选元素一定是主元素，例如[1, 2, 3]。

所以最后要对该候选元素进行验证，验证方式为重新遍历一遍数组，对该候选元素进行计数，最后判断该次数是否超过数组长度的一半。

程序

public class MajorityElem {
public int getMajorityElem(int[] nums) {
if (nums == null || nums.length == 0) {
return 0;
}

int candidate = nums[0];
int count = 1;

for (int i = 1; i < nums.length; i++) {
if (count == 0 || candidate == nums[i]) {
candidate = nums[i];
++count;
} else {
--count;
}
}

if (verify(nums, candidate)) {
return candidate;
}

return -1;
}

private boolean verify(int[] nums, int candidate) {
int count = 0;
for (int n : nums) {
if (n == candidate) {
++count;
}
}
return count > nums.length / 2;
}
}

Follow Up

Given an integer array of size n, find all elements that appear more than

⌊ n/3 ⌋

times. The algorithm should run in linear time and in O(1) space.

解决思路

思路与主元素问题相似，不同的在于需要设置两个candidates（数组中满足如上定义的元素个数最多为2个）。

程序

public class MajorityElem2 {
public List<Integer> majorityElement(int[] nums) {
List<Integer> res = new ArrayList<Integer>();
if (nums == null || nums.length == 0) {
return res;
}

int candicate1 = 0, candicate2 = 0;
int count1 = 0, count2 = 0;

for (int i = 0; i < nums.length; i++) {
if (count1 == 0 || candicate1 == nums[i]) {
++count1;
candicate1 = nums[i];
} else if (count2 == 0 || candicate2 == nums[i]) {
++count2;
candicate2 = nums[i];
} else {
--count1;
--count2;
}
}

verify(nums, candicate1, candicate2, res);

return res;
}

private void verify(int[] nums, int candicate1, int candicate2,
List<Integer> res) {
int len = nums.length;
int count1 = 0, count2 = 0;

for (int n : nums) {
if (n == candicate1) {
++count1;
}
if (n == candicate2) {
++count2;
}
}

if (count1 > len / 3) {
res.add(candicate1);
}

if (candicate1 == candicate2) {
return ;
}

if (count2 > len / 3) {
res.add(candicate2);
}
}
}

Bugs：

Case：[0, 0, 0], output: 0, 0

candidate1和candidate2有可能相同，如果都满足，只需要输出一个即可。

leetcode题目链接
https://leetcode.com/problems/majority-element/ https://leetcode.com/problems/majority-element-ii/