数组中占比超过一半的元素称之为主要元素。给你一个 整数 数组,找出其中的主要元素。若没有,返回 -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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-majority-element-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题目要求时间复杂度为 O(N), 最主要的空间复杂度为 O(1),所以不能用一开始想到的哈希表,只能选择投票法:
投票法的思路是这样的:遍历,每次删除两个不同的元素,直到投票过程无法继续,此时数组中剩下的要么是空,要么剩下的数都是一样的,虽然剩下的数都是一样的,但是也不一定就是主要元素,所以我们还得再遍历一次数组,判断是不是主要元素。
定义变量 candidate 和 count,然后循环遍历数组,当 count 等于 0 时,让 candidate 赋值为当前索引的元素,并让 count ++, 如果下一个元素不等于 candidate,说明两个元素不一样,此时思想是删除这两个元素,也就是让 count --,然后继续往后遍历,当两个元素相同时,count ++,当 count 减到 0 时,更新 candidate
class Solution {
public int majorityElement(int[] nums) {
int candidate = -1;
int count = 0;
for(int num : nums) {//循环遍历
if(count == 0) {//如果 count 为 0,就更新candidate
candidate = num;
}
if(candidate == num) {//相同元素,count ++
count++;
} else count--;
}
int length = nums.length;
int sum = 0;
for(int i : nums) {//计算出来的 candidate 不一定是主要元素
//需要再判断是不是超过数组的一半
if(i == candidate) {
sum++;
}
}
return sum > length / 2 ? candidate : -1;
}
}