题目描述:数组中占比超过一半的元素称之为主要元素。给你一个 整数 数组,找出其中的主要元素。若没有,返回 -1
。请设计时间复杂度为 O(N)
、空间复杂度为 O(1)
的解决方案。
示例 1:
输入:[1,2,5,9,5,9,5,5,5] 输出:5
示例 2:
输入:[3,2] 输出:-1
示例 3:
输入:[2,2,1,1,1,2,2] 输出:2
先抛开题目中的时间复杂度和空间复杂度的要求,是否可以用别的方法完成呢?既然是找主要元素,那就可能会有重复的元素,最先想到的方法是哈希表。
思路如下:先对数组进行升序排序找到哈希数组的大小,然后用哈希表统计数组中不同元素的个数,最后进行判断,元素次数超过数组长度的一半即为主要元素并返回该元素,否则没有主要元素,返回-1。
#C语言
int majorityElement(int* nums, int numsSize){
int cmp(const void *a , const void *b){
return *(int*)a > *(int*)b;
}
qsort(nums,numsSize,sizeof(int),cmp);
int *result = (int *)calloc(nums[numsSize-1]+1,sizeof(int));
int i;
for(i=0;i<numsSize;i++){
result[nums[i]] = result[nums[i]] + 1;
}
for(i=0;i<nums[numsSize-1]+1;i++){
if(result[i] > numsSize/2){
return i;
}
}
return 0;
}
上述代码,运行给出的示例可以通过,但是提交时报错,[-1,1,1,1,2,1],这个数组测试不通过。
我们忽略了一个问题,我们创建的哈希表,是从0开始的,也就是说,数组的元素都是大于等于零的,那对于小于零的元素怎么办呢?给出的示例没有小于零的元素,不代表题目中给出的数组没有小于零的元素。
如果还是以这个方法为基础进行修改,或许你会说,再创建一个哈希数组,用于存储小于零的元素,思路方法不变,最后进行统一判断,找出主要元素,但是提交时还会报错,对于数组[2147483647],nums[numsSize-1]+1这个代码片段会报错,因为2147483647+1已经超出了 int 类型所能表示的范围。
runtime error: signed integer overflow: 2147483647 + 1 cannot be represented in type 'int' [solution.c]
如果继续以这个方法为基础进行修改,已经没有意义了,这个方法原本就不符合题目中的要求,而且已经足够繁琐,再改下去,显然没意义。
此题的正确解法:Boyer-Moore 投票算法
Boyer-Moore 投票算法的基本思想是:一个变量candidate:表示元素,一个变量count:用于计数
过程如下:
如果count = 0,将数组元素赋给candidate
if(count == 0){
candidate = nums[i];
}
如果数组元素等于candidate,则count+1,否则count-1
if(candidate == nums[i]){
count++;
}
else{
count--;
}
遍历结束后,candidate的值表示一个元素,它可能是任意一个元素,也可能是主要元素,所以需要进行第二次遍历验证。
count = 0;
for(i=0;i<len;i++){
if(nums[i] == candidate){
count++;
}
}
Boyer-Moore 投票算法推理验证:为什么当数组中存在主要元素时,Boyer-Moore 投票算法可以确保得到主要元素。因为遇到相同的元素,count+1,遇到不同的元素,count-1。如果存在主要元素,那么主要元素的出现次数一定大于其他元素的出现次数之和,因此在遍历过程中,主要元素和其他元素两两抵消,最后一定剩下至少一个主要元素。
完整的Java代码和C代码如下:
class Solution {
public int majorityElement(int[] nums) {
int candidate = 0;
int count = 0;
int len = nums.length;
int i;
for(i=0;i<len;i++){
if(count == 0){
candidate = nums[i];
}
if(candidate == nums[i]){
count++;
}
else{
count--;
}
}
count = 0;
for(i=0;i<len;i++){
if(nums[i] == candidate){
count++;
}
}
return count * 2 > len ? candidate : -1;
}
}
int majorityElement(int* nums, int numsSize){
int candidate = -1;
int count = 0;
int i;
for(i=0;i<numsSize;i++){
if(count == 0){
candidate = nums[i];
}
if(candidate == nums[i]){
count++;
}
else{
count--;
}
}
count = 0;
for(i=0;i<numsSize;i++){
if(nums[i] == candidate){
count++;
}
}
return count*2>numsSize?candidate:-1;
}