力扣面试150题 ｜ 多数元素

题目描述

给定一个大小为 n 的数组 nums ，返回其中的多数元素。多数元素是指在数组中出现次数 大于 ⌊ n/2 ⌋ 的元素。

你可以假设数组是非空的，并且给定的数组总是存在多数元素。

示例 1：

输入：nums = [3,2,3]
输出：3

示例 2：

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

提示：

• n == nums.length
• 1 <= n <= 5 * 104
• -109 <= nums[i] <= 109

进阶：尝试设计时间复杂度为 O(n)、空间复杂度为 O(1) 的算法解决此问题。

解题思路

统计每个元素出现的次数，可以想到哈希表，这里用unordered_map，键存放元素，值存放元素出现的次数。

遍历数组，将每个元素都加入哈希表中，同时要维护最大的值。最后将最大值输出即可。

代码实现

class Solution {
public:
    int majorityElement(vector<int>& nums) {
		unordered_map<int, int> countNum;
		int result = 0, count = 0;
		for (int i : nums) {
			++countNum[i]; // 这里用前置递增是为了节省内存，后置递增也可以
			// 维护最大的值
			if (countNum[i] > count) {
				result = i;
				count = countNum[i];
			}
		}
		return result;
    }
};