【NO.5】LeetCode经典150题-169. 多数元素
【简单】 【HOT100】
给定一个大小为 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) 的算法解决此问题。
解题
方法一:哈希表
// 时间空间都为O(n)
class Solution {
public int majorityElement(int[] nums) {
// HashMap 存出现的次数
int n = nums.length;
HashMap<Integer, Integer> map = new HashMap<>();
for (int i = 0;i < n; i++) {
// 有,次数+1
if (map.containsKey(nums[i])) {
map.put(nums[i], map.get(nums[i]) + 1);
} else {
// 没有就放进入
map.put(nums[i], 1);
}
}
int pos = n/2;
int result = -1;
Set<Integer> keys = map.keySet();
for (Integer key : keys) {
if (map.get(key) > pos) {
result = key;
}
}
return result;
}
}
方法二:摩尔投票法
如果我们把众数(也就是多数)记为 +1,把其他数记为 −1,将它们全部加起来,显然和大于 0,从结果本身我们可以看出众数比其他数多。
摩尔投票法思路: count 初始化为 1。 当遇到与 cand_num(众数) 相同的数,则票数 count = count + 1,否则票数 count = count - 1。 当票数 count 为 0 时,更换候选人,并将票数 count 重置为 1。 遍历完数组后,cand_num 即为最终答案。
// 时间O(n) 空间O(1)
class Solution {
public int majorityElement(int[] nums) {
int n = nums.length;
int candNum = nums[0];
int count = 1;
for (int i = 1; i < n; i++) {
if (candNum == nums[i]) {
count++;
} else {
count--;
}
if (count == 0) {
candNum = nums[i];
count = 1;
}
}
return candNum;
}
}