前言
今天在刷LC时遇到了一道特别有意思的题目——多数元素,本来题目的难度只是easy
,但翻开它的相关题解发现可谓是层出不穷,而这里主要也想分享其中一个特别有意思的算法——Boyer Moore投票算法,没错,正是那个同时设计了BM字符串匹配算法的BM。
题目描述
给定一个大小为 n
的数组 nums
,返回其中的多数元素。多数元素是指在数组中出现次数 大于 ⌊ n/2 ⌋
的元素。
你可以假设数组是非空的,并且给定的数组总是存在多数元素。
示例 1:
输入:nums = [3,2,3]
输出:3
示例 2:
输入:nums = [2,2,1,1,1,2,2]
输出:2
提示:
- n == nums.length
- 1 <= n <= 5 * 104
- -109 <= nums[i] <= 109
进阶:尝试设计时间复杂度为 O(n)、空间复杂度为 O(1) 的算法解决此问题。
自解(排序+双指针)
起初的想法是先进行一次快速排序,将相同的元素都整理在了一起,整个数组形成了一段又一段的相同数区间,然后定义两个指针i
和j
,分别指向相同数区间的头部和尾部,nums[i] == nums[j]
时令j
指针不断向右移动,当j
指向了下一区间时,此时得到了这个相同数区间的长度即为j - i
,然后判断j - i
是否大于n / 2
即可得出答案。
class Solution {
public int majorityElement(int[] nums) {
int i = 0, j = 0, n = nums.length;
Arrays.sort(nums);
while(j < n){
if(j == n - 1 *|| j - i > (n / 2)) return nums[i];
if(nums[i] == nums[j]) j ++;
else{
if(j - i > (n / 2)) return nums[i];
i = j;
}
}
return -1; //这段无论如何都不会被执行
}
}
这样做法的复杂度为 O(nlogn + n) = O(nlogn)。
但经过仔细一想才发现双指针这一步纯属多余,因为多数元素对应的数字一定是占据了数组一半以上的位置,所以在nums
排序后的中间位置一定会是该数。对应如下:
class Solution {
public int majorityElement(int[] nums) {
Arrays.sort(nums);
return nums[nums.length / 2];
}
}
Boyer-Moore投票算法
先来看官方给出的该算法定义:
一开始看时我是一脸直接懵QAQ,还好看了以下的算法通俗解释才恍然大悟,无不惊叹这简直媲美于天才般的想法!
这是我目前看到的最直观形象的解法,时间复杂度O(n),空间复杂度O(1)。
“同归于尽消杀法” :
由于多数超过50%, 比如100个数,那么多数至少51个,剩下少数是49个。
-
第一个到来的士兵,直接插上自己阵营的旗帜占领这块高地,此时领主 winner 就是这个阵营的人,现存兵力 count = 1。
-
如果新来的士兵和前一个士兵是同一阵营,则集合起来占领高地,领主不变,winner 依然是当前这个士兵所属阵营,现存兵力 count++;
-
如果新来到的士兵不是同一阵营,则前方阵营派一个士兵和它同归于尽。 此时前方阵营兵力count --。(即使双方都死光,这块高地的旗帜 winner 依然不变,因为已经没有活着的士兵可以去换上自己的新旗帜)
-
当下一个士兵到来,发现前方阵营已经没有兵力,新士兵就成了领主,winner 变成这个士兵所属阵营的旗帜,现存兵力 count ++。
-
就这样各路军阀一直以这种以一敌一同归于尽的方式厮杀下去,直到少数阵营都死光,那么最后剩下的几个必然属于多数阵营,winner 就是多数阵营。(多数阵营 51个,少数阵营只有49个,死剩下的2个就是多数阵营的人)。
如此一来算法的时间复杂度降到了 O(n)。
进一步,代码实现如下:
class Solution {
public int majorityElement(int[] nums) {
int winner = nums[0]; //定义的第一个领主
int count = 1;
for(int i = 0;i < nums.length;i ++){
if(nums[i] == winner) count ++; //加入阵营
else count --; //相杀,同归于尽
if(count == 0){
winner = nums[i]; //易主
count ++;
}
}
return winner; //战斗到最后还剩兵力的一定是那个“多数的阵营”
}
}
算法还尚未有理论的证明,但确是能经过直觉与思考明白它是可行的。