LeetCode - 算法

Swift - LeetCode - 多数元素

2022-08-06  本文已影响0人  晨曦的简书

题目

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

示例 1:

  • 输入:nums = [3,2,3]
  • 输出:3

示例 2:

  • 输入:nums = [2,2,1,1,1,2,2]
  • 输出:2

方法一:哈希表

思路及解法

我们知道出现次数最多的元素大于 \lfloor \dfrac{n}{2} \rfloor 次,所以可以用哈希表来快速统计每个元素出现的次数。

我们使用哈希映射(HashMap)来存储每个元素以及出现的次数。对于哈希映射中的每个键值对,键表示一个元素,值表示该元素出现的次数。

我们用一个循环遍历数组 nums 并将数组中的每个元素加入哈希映射中。在这之后,我们遍历哈希映射中的所有键值对,返回值最大的键。我们同样也可以在遍历数组 nums 时候使用打擂台的方法,维护最大的值,这样省去了最后对哈希映射的遍历。

代码

class Solution {
    func majorityElement(_ nums: [Int]) -> Int {
        var counts: [Int : Int] = [:]
        var majority: Int = 0
        for num in nums {
            var cnt: Int = counts[num] ?? 0
            cnt += 1
            if cnt > (nums.count / 2) {
                majority = num
                break
            }
            counts[num] = cnt
        }
        return majority
    }
}

复杂度分析

方法二:排序

思路及解法

如果将数组 nums 中的所有元素按照单调递增或单调递减的顺序排序,那么下标为 \lfloor \dfrac{n}{2} \rfloor 的元素(下标从 0 开始)一定是众数。

对于这种算法,我们先将 nums 数组排序,然后返回上文所说的下标对应的元素。下面的图中解释了为什么这种策略是有效的。在下图中,第一个例子是 n 为奇数的情况,第二个例子是 n 为偶数的情况。

对于每种情况,数组下面的线表示如果众数是数组中的最小值时覆盖的下标,数组下面的线表示如果众数是数组中的最大值时覆盖的下标。对于其他的情况,这条线会在这两种极端情况的中间。对于这两种极端情况,它们会在下标为 \lfloor \dfrac{n}{2} \rfloor 的地方有重叠。因此,无论众数是多少,返回 \lfloor \dfrac{n}{2} \rfloor 下标对应的值都是正确的。

代码

class Solution {
    func majorityElement(_ nums: [Int]) -> Int {
        var tempNums = nums
        tempNums.sort()
        return tempNums[tempNums.count / 2]
    }
}

复杂度分析

上一篇 下一篇

猜你喜欢

热点阅读