力扣面试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
,键存放元素,值存放元素出现的次数。
遍历数组,将每个元素都加入哈希表中,同时要维护最大的值。最后将最大值输出即可。
代码实现
cpp
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;
}
};