Java实现 LeetCode 229 求众数 II（二）

229. 求众数 II

给定一个大小为 n 的数组，找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。

说明: 要求算法的时间复杂度为 O(n)，空间复杂度为 O(1)。

示例 1:

输入: [3,2,3]
输出: [3]
示例 2:

输入: [1,1,1,3,3,2,2,2]
输出: [1,2]

class Solution {
       public List<Integer> majorityElement(int[] nums) {
         List<Integer> res = new ArrayList<>();
        if (nums.length < 1) {
            return res;
        }
        int num1 = nums[0];
        int count1 = 0;
        int num2 = nums[0];
        int count2 = 0;

        for (int i = 0; i < nums.length; i++) {
            int temp = nums[i];
            if (temp == num1) {
                count1++;
            } else if (temp == num2) {
                count2++;
            } else if (count1 == 0) {
                count1 = 1;
                num1 = temp;
            } else if (count2 == 0) {
                count2 = 1;
                num2 = temp;
            } else {
                count1--;
                count2--;
            }
        }
        count1 = 0;
        count2 = 0;
        int numSum = nums.length / 3;
        for (int i = 0; i < nums.length; i++) {
            int temp = nums[i];
            if (temp == num1) {
                count1++;
            } else if (temp == num2) {
                count2++;
            }
        }
        if (count1 > numSum) {
            res.add(num1);
        }
        if (num1 != num2 && count2 > numSum) {
            res.add(num2);
        }
        return res;
    }
}