17.10主要元素Java

17.10主要元素Java

题目描述

数组中占比超过一半的元素称之为主要元素。给定一个整数数组，找到它的主要元素。若没有，返回-1。
在时间复杂度为 O(N)，空间复杂度为 O(1) 内完成

输入输出样式

输入：[1,2,5,9,5,9,5,5,5]
输出：5

输入：[3,2]
输出：-1

输入：[2,2,1,1,1,2,2]
输出：2

本文题来自LeetCode：https://leetcode-cn.com/problems/find-majority-element-lcci/

思路

1. 刚开始想要用排序将数组排序，然后直接遍历一半的数组，检验当前元素nums[i] == nums[ i + nums.length / 2]，但是时间复杂度为O(排序)+O(n / 2)，超时了，遂放弃。
2. 后来看到了摩尔投票法，感叹真是巧妙，思路非常清晰简单。一个数组的主要元素一定是唯一的，且数量多于数组除去主要元素剩下的元素。那么就让元素间相互消耗：遍历一边数组，任意删除两个不同的数，直到只剩下相同的数，也就是铲除了非主要元素的其他元素。
3. 再解释一下：假设桶里一筐硬币(一毛，一角，一块)，一块硬币的数量大于一半。每拿出去一个一毛或者一角，就再拿出去一个一块的硬币，这样最后一毛的硬币和一角的硬币拿完了，框里就只会剩下一块的硬币。
4. 算法思想：candidate为主要元素(默认值随意设置)，count为数量(默认值为0)；遍历数组，从第一个元素x开始前，先验证count是否为0，为零则说明数量为零，则将x赋值给candidate；接着判断x == candidate ？，真则count++；否则count–；遍历完数组candidate就是主要元素
5. 但是摩尔投票法默认数组是存在主要元素的，但是对于形如[3,2,1]是没有主要元素的；因此在步骤4完成后要遍历一次数组，判断canditate是否是主要元素

算法分析

1. 遍历了两次数组，时间复杂度为O(n)
2. 定义了若干临时变量，空间复杂度为O(1)

摩尔投票函数

public int majorityElement(int[] nums) {
        int count = 0;
        int cnum = 0; //检验选出的candidate出现次数
        int candidate = 0;
        for (int i = 0; i < nums.length; i++) {
            if (count == 0) {
                candidate = nums[i];
            }
            if (nums[i] == candidate) count++;
            else count--;
        }
        for (int j = 0; j < nums.length; j++) {
            if (nums[j] == candidate) cnum++;
        }
        if (cnum <= nums.length / 2) candidate = -1;//检查
        return candidate;
        }
}

在主函数中调用转置函数进行测试

class Solution1710 {
    public static void main(String[] args) {
        int[] num = new int[]{2, 7, 3, 5, 5};
        Solution1710 sol = new Solution1710();
        int rst = sol.majorityElement(num);
        System.out.println(rst);
    }

    public int majorityElement(int[] nums) {...}
}

输出结果

其他解法

主要元素还可以使用分治法，位运算等。其实这里的摩尔投票就有分治的思想：将大的问题逐渐变小：如在[1,2,1,1]中找主要元素，根据摩尔投票。第一位和第二位不相同，那么相当于会变成在[1,1]中寻找主要元素