高频面试算法之求众数

1、简介

求众数在面试考察算法环节，非常常见，解法较多，而且还有扩展，笔者在京东科技的面试中曾被考察到该题目。

题目：给定一个大小为 n 的数组，找到其中的多数元素。多数元素是指在数组中出现次数 大于 ⌊ n/2 ⌋ 的元素。你可以假设数组是非空的，并且给定的数组总是存在多数元素。

2、解法一

分析该题目，最常见的思路即使用hash表，求得每个元素的个数，再找到其中最大元素个数即可。时间复杂度：O(n)；空间复杂度：O(n)。因为空间复杂度较高，所以一般情况下面试官还会继续追问可否优化空间复杂度。

2、解法二

分析该题目不难发现：将数组排序后，中间那个数一定是众数。

class Solution {
    public int majorityElement(int[] nums) {
        Arrays.sort(nums);
        return nums[nums.length / 2];
    }
}

时间复杂度：O(nlog n)，空间复杂度O(logn)。看题解，这个解法显然也不是面试官想要的。

3、解法三

摩尔投票。如果我们把众数记为 +1+1，把其他数记为 -1−1，将它们全部加起来，显然和大于 0，从结果本身我们可以看出众数比其他数多。

简单理解：假设现在有两队士兵交战，规则是必须一比一的消耗，即只能同归于尽。消耗到最后，留下的士兵即众数。

class Solution {
    public int majorityElement(int[] nums) {
        int count = 0;
        // 候选数
        Integer cadicate = null;

        for(int num:nums){
            // 如果是初始条件，或者双方消耗完了，需要再次选择一个候选数
            if(count == 0){
                cadicate = num;
            }
            // 开始消耗。若候选众数与当前数相同，则加一；否则，同归于尽，即减一
            count = num == cadicate ? ++count:--count;
        }
        
        return cadicate;
    }
}

时间复杂度O(n)；空间复杂度O(1)。一般答到这里，即可通过。如果没有刷过算法题，很难想到这种解法。

4、扩展

题目：给定一个大小为 n 的整数数组，找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。
整体思路与第一套题的思路相同，第一题众数是一个；第二题众数是两个。依然采取摩尔投票的策略即可。解完第一题一定要看第二题，因为面试官极有可能会问，反正我是遇到了。。。。

class Solution {
    // 摩尔投票的改进版
    // 根据题意，求超过 n/3 的数，则满足这样条件的数，最多只有两个
    // 从数组中挑选出两个数，记做 n1，n2，做摩尔投票
    // 如果当前数与n1，n2都不同，则放弃投票（可以理解为三个数消消乐）
    public List<Integer> majorityElement(int[] nums) {
        int n1 = nums[0], c1 = 0;
        int n2 = nums[0], c2 = 0;

        for (int i = 0; i < nums.length; i++) {
            int num = nums[i];

            if (n1 == num) {
                c1++;
                continue;
            }

            if (n2 == num) {
                c2++;
                continue;
            }

            if (c1 == 0) {
                n1 = num;
                c1++;
                continue;
            }

            if (c2 == 0) {
                n2 = num;
                c2++;
                continue;
            }

            c1--;
            c2--;
        }

        int count1 = 0, count2 = 0;
        for (int num : nums) {
            if (num == n1) {
                count1++;
            } else if (num == n2) {
                count2++;
            }
        }

        List<Integer> list = new ArrayList<>();
        if (count1 > nums.length / 3) {
            list.add(n1);
        }

        if (count2 > nums.length / 3) {
            list.add(n2);
        }

        return list;
    }
}

5、原题链接

第一题
第二题