leetcode 1838. 最高频元素的频数

最新推荐文章于 2025-09-22 22:06:16 发布

原创最新推荐文章于 2025-09-22 22:06:16 发布 · 352 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

Leetcode

本文介绍如何通过排序和滑动窗口算法解决给定整数数组nums和整数k的问题，目标是在k次操作后找到最高频元素的最大可能频数。实例演示了如何应用这一思路来优化数组元素分布并求解最优解。

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

给你一个整数数组 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 <= 10^5
1 <= nums[i] <= 10^5
1 <= k <= 10^5

思路：排序+双指针（滑动窗口思想)
1、从小到大排序
2、排序过后，从小到大，逐次上升，想象成具有层次感的一排树木，如果想要一个区间 [ l, r ] 之间的数全部变为 nums[ r ], 首先求这些元素的和 sum，那么需要+1的次数就 = 全部变为nums[ r ]之后的总和 - 改变之前的总和，即nums[ r ] * ( r - l + 1) - sum, 如果这个结果 <= k ，窗口扩大，r + 1；否则，窗口缩小，l + 1，更新 sum.

class Solution {
public:
    int maxFrequency(vector<int>& nums, int k) {
        int n = nums.size();
        if(n == 1) return 1;
        sort(nums.begin(), nums.end()); 
        int l = 0;     //左指针
        int r = 1;     //右指针
        //注意变量类型
        long long sum = nums[0];
        int res = 1;
        while(r < n){
            sum += nums[r];
            if((long long)(r - l + 1) * nums[r] - sum <= k){
                res = max(res, r - l + 1);
            }
            else{
                sum -= nums[l];
                l++;
            }
            r++;
        }
        return res;
    }
};