LeetCodeDiary

A Diary for solving LeetCode problems

View on GitHub

1838. 最高频元素的频数

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

给你一个整数数组 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

提示:

滑动窗口 中等

代码

class Solution:
    def maxFrequency(self, nums: List[int], k: int) -> int:
        n = len(nums)
        nums.sort()
        total, l, res = 0, 0, 1
        # 滑动方框,不断右移右边框
        for r in range(1,n):
            # 计算要添加的格子数
            # 注意这里乘的是r-l!
            total += (nums[r] - nums[r-1]) * (r-l)
            while total > k:
                # 超过范围,尝试左移左边框
                # 减去多的框就可以继续循环
                total -= nums[r] - nums[l]
                l += 1
            # 每次右移后比对当前框和之前框那个包围的横坐标更大
            res = max(res,r-l+1)
        return res