数组中占比超过一半的元素称之为主要元素。给你一个 整数 数组,找出其中的主要元素。若没有,返回 -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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
首先想到的是用map<数组元素,对应个数>。能通过,但是时间花了18ms。
class Solution {
public int majorityElement(int[] nums) {
int length = nums.length;
Map<Integer,Integer> map = new HashMap<Integer,Integer>();
for (int i = 0; i < length; i++) {
if (map.containsKey(nums[i])) {
map.put(nums[i], map.get(nums[i])+1);
}else {
map.put(nums[i], 1);
}
if (map.get(nums[i])>length/2) {
return nums[i];
}
}
return -1;
}
}
然后想到这是个统计众数的题目(最后要确认一下众数个数是不是超过总元素数目的一半),统计众数的方法:摩尔投票法。
摩尔投票法大致为:一个k记录个数,一个cand记录众数候选者。
遍历整个数组,
遇到k=0的时候,给cand赋上现在数组的元素,k=1;
如果k不是0,看看现在的元素是不是cand,是就k++,不是就k–。
(原理是:如果相同,个数就加一,这个好理解。但是目前的候选元素和当前遍历到的元素不同的时候,发生冲突了,得解决啊,所以他俩各退一步,当前候选者也就个数减一了)。
那会不会发生当前的cand遇到的竞争者太多了,k变成负数的情况呢?
这个是不会的,因为有判断k=0的时候,cand重新赋值,也就是新的候选者诞生的情况。
class Solution {
public int majorityElement(int[] nums) {
int k = 0;
int cand = 0;
for (int i : nums) {
if (k == 0) {
cand = i;
k = 1;
}else if(cand == i){
k++;
}else {
k--;
}
}
k = 0;
for (int i : nums) {
if (cand == i) {
k++;
}
}
if (k > nums.length/2) {
return cand;
}else {
return -1;
}
}
}