LeetCode 每日 1 题:最高频元素的频数

37 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第 26 天,点击查看活动详情

最高频元素的频数

原题地址

元素的 频数 是该元素在一个数组中出现的次数。

给你一个整数数组 nums 和一个整数 k 。在一步操作中,你可以选择 nums 的一个下标,并将该下标对应元素的值增加 1

执行最多 k 次操作后,返回数组中最高频元素的 最大可能频数

示例 1:

输入:nums = [1,2,4], k = 5
输出:3
解释:对第一个元素执行 3 次递增操作,对第二个元素执 2 次递增操作,此时 nums = [4,4,4] 。
4 是数组中最高频元素,频数是 3

示例 2:

输入:nums = [1,4,8,13], k = 5
输出:2
解释:存在多种最优解决方案:
- 对第一个元素执行 3 次递增操作,此时 nums = [4,4,8,13] 。4 是数组中最高频元素,频数是 2 。
- 对第二个元素执行 4 次递增操作,此时 nums = [1,8,8,13] 。8 是数组中最高频元素,频数是 2 。
- 对第三个元素执行 5 次递增操作,此时 nums = [1,4,13,13] 。13 是数组中最高频元素,频数是 2

示例 3:

输入:nums = [3,9,6], k = 2
输出:1

提示:

  • 1<=nums.length<=1051 <= nums.length <= 10^5
  • 1<=nums[i]<=1051 <= nums[i] <= 10^5
  • 1<=k<=1051 <= k <= 10^5

思路分析

  1. 首先将数组进行升序排列,定义左指针,初始指向是 0,然后定义 result 初始值为 1;
  2. 移动右指针进行遍历,更新 total 值,如果 total 值比 k 大,则需要减小 total,通过向左移动左指针来降低;
  3. 然后在 total 允许的范围内,更新最高的频数,也就是左右两个指针的距离;
  4. 最后返回 result 值即可。

AC 代码

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var maxFrequency = function(nums, k) {
    nums.sort((a, b) => a - b)
    
    let total = 0
    let left = 0
    let result = 1
    for (let right = 1; right < nums.length; right++) {
        total += (nums[right] - nums[right - 1]) * (right - left)
        
        while (total > k) {
            total -= nums[right] - nums[left]
            left++
        }
        
        result = Math.max(result, right - left + 1)
    }
    return result
};

结果:

  • 执行结果: 通过
  • 执行用时:220 ms, 在所有 JavaScript 提交中击败了67.65%的用户
  • 内存消耗:55.1 MB, 在所有 JavaScript 提交中击败了41.18%的用户
  • 通过测试用例:71 / 71

END