你听说过摩尔投票法吗

Posted 2021-11-24 战场小包

前言

前几天在刷力扣题目时，看官方题解发现的算法，我感觉这个算法非常有意思，是一个很好的思维开拓，因此我整理摩尔投票算法的一些知识，分享给大家。

摩尔投票法

博耶-摩尔多数投票算法( Boyer–Moore majority vote algorithm ),中文常作多数投票算法、摩尔投票算法等，是一种用来寻找一组元素中占多数元素的常数空间级时间复杂度算法。

算法思想

在集合中寻找可能存在的多数元素，这一元素在输入的序列重复出现并占到了序列元素的一半以上；在第一遍遍历之后应该再进行一个遍历以统计第一次算法遍历的结果出现次数，确定其是否为众数；如果一个序列中没有占到多数的元素，那么第一次的结果就可能是无效的随机元素。

过程可以分为两个阶段：

• 投票阶段: 即投票人之间票数进行抵消
• 计数阶段: 计算对抗结果中最后剩下的那个候选人票数是否有效。

举个栗子

上面的文字属实有几分枯燥无味，下面咱们来举一个有趣的栗子：

斯巴达勇士们经过了无数年的奋斗，成立了斯巴达共和国，走上了社会主义道路。

今天是斯巴达第一届人民代表大会，会议将通过全民选举的方式选出第一届共和国主席。为了保证主席的权威性，只有得票数过半才能成为新共和国的领袖。

但现在斯巴达几百万人口，统计选票是个大问题啊。而且我斯巴达勇士勇猛无敌，岂能浪费在算数上。会议组织人绞尽脑汁，终于想出了一个绝妙的点子，不仅可以选出共和国领袖，也可以展现斯巴达共和国的勇武。

规则是这样的：

1. 如果两个勇士选举意见不合，就去场外打一架，投票阶段就不要回来了。所有选举意见不合的都去激情对撞了，剩下的斯巴达勇士(可能有多个)意见肯定统一，他们的选举对象有可能就是最后的主席。
2. 但如果光靠剩下的人判断领袖是存在漏洞的，例如 [1,2,2,1,3,3] 这种，前面两个候选人选票互相内耗了，第三个候选人也没有达到总票数的一般，因此还需要计数阶段，统计一下最后所剩候选人的票数是否达到总票数的一半。

图文讲解

看完阿包举的例子如果还不懂，咱们再来看个图文的。

思路

根据上面的算法思想，我们将当前票数最多的候选人与获得的票数(抵消后)分别存储在 major 和 count 中。

当我们遍历下一个选票时，判断当前 count 是否为零：

• 如果 count == 0 : 代表 major 空缺，直接将当前候选人赋值给 major，并将 count ++
• 若 count != 0 : 代表当前 major 的票数未被完全抵消，令 count --，不同意见的斯巴达勇士去激情对撞。

初始值 count = 0, major = -1

详细图解
以 [2,2,1,3,1,2,2] 为例。

遍历数组第一个元素 2，因 count == 0，所以将 2 赋值给 major，且票数 count = 1

第二个数组元素依旧是候选人 2，得票数 count ++

第三个元素是 1，与 major 冲突，因此发生激情对撞，当前 major 的票数抵消 1 票。

第四个元素是 3， 与 major 冲突，产生激情对撞，当前 major 的票数抵消一票。

当遍历到第五个元素 1 时，此时 count = 0，说明 marjor 位置空缺，所以令 majro = 1，且 count = 1

第六个元素是 2，与 major 冲突，产生激情对撞，当前 major 的票数抵消 1 票，此时 count 又变为 0

遍历最后一个元素 2，当前 count = 0，说明 marjor 位置空缺，所以令 majro = 1，且 count = 1

遍历完毕，2 元素很有可能就是斯巴达新的领袖。

计数阶段: 统计 2 元素出现的次数是否大于一半，2 出现四次，大于一半，所以我宣布斯巴达共和国未来的领袖就是 2 了(怎么听起来怪怪的。。。)

真题实战

真题一：主要元素

题目来源: 面试题 17.10. 主要元素

题目描述：数组中占比超过一半的元素称之为主要元素。给你一个 整数 数组，找出其中的主要元素。若没有，返回 -1 。请设计时间复杂度为 O(N) 、空间复杂度为 O(1) 的解决方案。

如果没有最后一句要求，肯定咱们大脑中第一瞬间想到的就是哈希表，使用哈希表存储每个元素的出现次数。

但是咱们现在可不是凡人啊，连斯巴达共和国的领袖都可以选举出来了，肯定要上摩尔选举法啊。

这个题的思想与我举得图文讲解极度类似，我就不多做赘述了。

/**
 * @param {number[]} nums
 * @return {number}
 */
var majorityElement = function(nums) {
    let major = -1;
    let count = 0;
    for (var i = 0; i<nums.length; i++) {
        if (count === 0) {
            major = nums[i];
        }
        if (nums[i] === major) {
            count++;
        } else {
            // 票数抵消
            count--;
        }
    }
    count = 0;
    const length = nums.length;
    // 计数阶段
    for (var i = 0; i<nums.length; i++) {
        if (nums[i] === major) {
            count++;
        }
    }
    return count * 2 > length ? major : -1;
};

真题二：求众数 II

题目来源: 229. 求众数 II

题目描述：给定一个大小为 n 的整数数组，找出其中所有出现超过 ⌊ n/3 ⌋ 次的元素。请设计时间复杂度为 O(N) 、空间复杂度为 O(1) 的解决方案。

我当时刷这道题时苦思冥想，咋都想不出空间复杂度为 O(1)的解决方案，看官方题解醍醐灌顶，这也能摩尔投票法。咱们来一起分析一下：

做这个题之前，首先要理解：出现超过 ⌊ n/3 ⌋ 次的元素最多有 2 个。咱们可以稍微反证一下：如果出现了 2 个以上超过 ⌊ n/3 ⌋ 次的元素，例如 3 个，哪意味着当前数组中会存在 3 * ⌊ n/3 ⌋ > n，与现实冲突。

思路

我把这个题形象一下，其实就相当于斯巴达勇士们要选正副领袖，所以现在两个不同意见得勇士打架就不合理了，因为这两个勇士的候选人可能是正副领袖。

既然是要选两个领袖，那是不是也可以类比为三个不同意见得勇士打架抵消那，下面我们来具体操作一下，看看能不能实现：

如果数组中只有一个元素 x 超过了 ⌊ n/3 ⌋，把数组分为两部分：一部分为 k 个 x，另一部分为 (n-k)/3 组三个不同的元素，如果三个不同元素会被抵消，最终只会剩下 k 个 x。

如果数组中有两个元素 x,y 超过了 ⌊ n/3 ⌋，把数组分为三部分：第一部分为 k 个 x，第二部分为 l 个 y，第三部分为 (n-k-l) / 3 组三个不同的元素，如果三个不同元素会被抵消，最终会剩下 k 个 x 和 l 个 y。

通过上面得分析，三个不同意见的勇士相抵消是可以实现的，算法流程为：

• 检查当前元素是否为第一个选中的元素或第二个选中的元素。如果存在相同，对应元素得票数加一。
• 如果与两个元素都不相同，则三个不同得元素抵消一次
• 抵消结束后，若存在最终选票大于 0 得元素，进行计数阶段，检查该元素次数是否大于 ⌊ n/3 ⌋

上代码：

/**
 * @param {number[]} nums
 * @return {number[]}
 */
var majorityElement = function(nums) {
    let major1 = 0,
        major2 = 0,
        count1 = 0,
        count2 = 0;
    const length = nums.length;
    // 投票抵消阶段
    for (let i = 0; i<length; i++) {
        if (count1 > 0 && major1 === nums[i]) {
            count1 ++;
        }else if (count2 > 0 && major2 === nums[i]) {
            count2 ++;
        }else if (count1 === 0) {
            count1 ++;
            major1 = nums[i];
        }else if (count2 === 0) {
            count2 ++;
            major2 = nums[i];
        }else {
            count1 --;
            count2 --;
        }
    }
    // 计数阶段
    let cnt1 = 0, cnt2 = 0, res = [];
    for (let i = 0; i<length; i++) {
        if (count1 > 0 && major1 === nums[i]) {
            cnt1 ++;
        }
        if (count2 > 0 && major2 === nums[i]) {
            cnt2 ++;
        }
    }
    if (count1 > 0 && cnt1 > Math.floor(length / 3)) {
        res.push(major1);
    }
    if (count2 > 0 && cnt2 > Math.floor(length / 3)) {
        res.push(major2);
    }
    return res;
};

往期精彩文章

• 牛客最新前端JS笔试百题
• JavaScript之彻底理解原型与原型链
• JavaScript之预编译学习
• JavaScript之彻底理解EventLoop
• 《2w字大章 38道面试题》彻底理清JS中this指向问题

参考博客: 图解摩尔投票法（Python & Go）