LeetCode169、多数元素

LeetCode 系列记录我学习算法的过程。

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第27天，点击查看活动详情

题目

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

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

示例：

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


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

提示：

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

思路

多数元素指的是在一个数组里出现次数占数组长度一半以上的数组项

那就可以通过记录每个数值出现的次数，当出现次数超过数组长度的一半时，将其返回即可：

• 首先我们定义一个变量 c 获取数组长度的一半
• 然后定义一个 map 集合用来存储每个数字出现的次数
• 然后遍历数组，更新当前项出现的次数
• 当出现次数超过数组长度一半时，返回当前项

代码实现

/**
 * @param {number[]} nums
 * @return {number}
 */
var majorityElement = function(nums) {
    // 首先获取数组长度的一半
    let c = nums.length / 2
    // 定义一个map集合用来统计每个数字出现的次数
    let map = new Map()
    // 遍历数组
    for(let i in nums) {
        // 在map集合里更新当前数字出现的次数
        map.set(nums[i], (map.get(nums[i]) || 0) + 1)
        // 当次数超过 c 时，将当前项返回
        if(map.get(nums[i]) > c) 
            return nums[i]
    }
};

优化

还有一种非常巧妙的方法：

既然多数元素是出现次数一半以上的元素，那么只要将不同元素两相抵消，剩下还存在的就肯定是多数元素了

var majorityElement = function(nums) {
    // 首先将第一个元素记录为多数元素，出现次数为 1
    let temp = nums[0], time = 1

    // 从第二项开始遍历
    for(let i = 1; i < nums.length; i++) {
        // 当前项和记录的多数元素相同时，出现次数+1
        if(nums[i] === temp) {
            time++
        } else {
            // 不同时进行抵消，次数-1
            time--
            // 当抵消完后，将下一个元素记录为多数元素，并重置出现次数
            if (time === 0) {
                temp = nums[i + 1]
                time = 1
                i++
            }
        }
    }
    // 最后返回记录的结果即可
    return temp
};