1838. 最高频元素的频数

Difficulty
Medium
Tags
滑动窗口
URL
https://leetcode.cn/problems/frequency-of-the-most-frequent-element/
Star
元素的 频数 是该元素在一个数组中出现的次数。
给你一个整数数组 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 <= 105
  • 1 <= nums[i] <= 105
  • 1 <= k <= 105
通过次数32,570提交次数75,918

滑动窗口

思路
题目说最多执行k次操作。那暴力的方法就是 遍历每一个数字，考虑最多执行k次操作让该数字称为频率最高的。复杂度为O(n^2)。
抽象为：将数组排序后，找一个最大窗口，使得该窗口内的元素值相同。
有序数组可能好一点，因为有序数组不用考虑后面的元素，只需要考虑前面的。再进一步，有序素组前面的数字肯定小于等于后面的元素。所以在遍历到 nums[r]时，因为要使得 nums[r]为频率最大的，此时可以用到 nums[r-1]的情况。
 
题解
class Solution:
    def maxFrequency(self, nums: List[int], k: int) -> int:
        nums.sort()
        left, right = 0, 1
        res = 1
        wds = 0
        while right < len(nums):
            # 将 nums[l:r] 中的数字都改为 nums[r] 需要的修改次数
            # 因为在前面的过程中，已经将 nums[l:r-1] 中的数字都改为 nums[r-1]了，所以只需要在所有数字的基础上增加 nums[r] - nums[r-1]
            wds += (nums[right] - nums[right-1]) * (right-left)
            while wds > k:
                wds -= nums[right] - nums[left]
                left += 1
            right += 1
            res = max(res, right-left)

        return res