LeetCode--面试题 17.10. 主要元素

Problem: 面试题 17.10. 主要元素

题目

数组中占比超过一半的元素称之为主要元素。给你一个整数数组，找出其中的主要元素。若没有，返回 -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

思路

算法的策略是从前向后扫描数组元素，标记出一个可能成为主元素的元素Num。然后重新计数，确认Num是否是主元素。

解题方法

算法可分为以下两步：

1. 选取候选的主元素：
   依次扫描所给数组中的每个整数，将第一个遇到的整数Num保存到c中，记录Num的出现次数为1；若遇到的下一个整数仍等于Num，则计数加1，否则计数减1；当计数减到0时，将遇到的下一个整数保存到c中，计数重新记为1，开始新一轮计数，即从当前位置开始重复上述过程，直到扫描完全部数组元素。
2. 判断c中元素是否是真正的主要元素：
   再次扫描数组，统计c中元素出现的次数，若大于n/2，则为主元素；否则，序列中不存在主元素。

复杂度

• 时间复杂度:

$O(n)$

• 空间复杂度:

$O(1)$

Code

class Solution {
public:
    int majorityElement(vector<int>& nums) {
        //扫描得出可能是主元素的元素c
        int n = nums.size();
        int c,count;
        c = nums[0];
        count = 1;
        for(int i = 1; i < n; i++) {
            if(nums[i]==c) count++;
            else {
                if(count>0) count--;
                else {
                    c = nums[i];
                    count = 1;
                }
            }
        }
        //确认是否是主元素
        for(int j = count = 0; j < n; j++) {
            if(nums[j] == c) count++;
        }
        if(count>n/2) return c;
        return -1;
    }
};