面试题 17.10. 主要元素（简单）- LeetCode

题目描述

自己解法

解法一：先给列表排序，在排序完的数组中，下标为数组长度/2的元素有可能是主元素，再统计数组中该元素的个数，若超过一半则证明是主元素。

class Solution:
    def majorityElement(self, nums: List[int]) -> int:
        nums = sorted(nums)
        element = nums[len(nums) // 2]
        count = 0
        for val in nums:
            if val == element:
                count += 1
        return element if count > len(nums) / 2 else -1

时间复杂度：$O(nlogn)$ 空间复杂度：$O(n)$

解法二：开一个字典来存储所有元素出现的次数，进行统计即可：

class Solution:
    def majorityElement(self, nums: List[int]) -> int:
        ele_dict = {}
        for val in nums:
            if val in ele_dict.keys():
                ele_dict[val] += 1
            else:
                ele_dict[val] = 1
        for key in ele_dict.keys():
            if ele_dict[key] > len(nums) / 2:
                return key
        return -1

时间复杂度：$O(n)$ 空间复杂度：$O(n)$

题解区解答

使用摩尔投票法，具体代码参考题解区解法。时间复杂度$O(n)$，空间复杂度$O(1)$

class Solution {
    public int majorityElement(int[] nums) {
        // 投票算法
        int temp = nums[0];
        int count = 1;
        for (int i = 1; i < nums.length; i++) {
            if (nums[i] == temp) {
                count++;
            } else {
                count--;
            }
            if (count == 0) {
                temp = nums[i];
                count = 1;
            }
        }

        // 上面代码找到的是众数，不一定满足题目要求，需进行验证
        int t = nums.length / 2 + 1;
        count = 0;
        for (int num : nums) {
            if (num == temp) count++;
            if (count == t) return temp;
        }
        return -1;
    }
}