LeetCode 1838. 最高频元素的频数

难度：中等。
标签：数组，二分查找，前缀和，滑动窗口。

用最简单的思路做，先排序，然后从大往小遍历，查看当前数是否是最大频数的数字。

正确解法：

class Solution {
public:
    int maxFrequency(vector<int>& nums, int k) {
        int n = nums.size();
        sort(nums.begin(), nums.end());
        int ans = 0;
        for(int i = n - 1; i >= 0; --i){
            if(i < ans)break;
            int now_num = 1;
            int now = nums[i];
            int j = i - 1;
            for(; j >= 0; --j){
                if(nums[j] != now)break;
                now_num++;
            }
            i = j + 1; 
            int change_time = k;
            for(; j >= 0; --j){
                if(change_time < now - nums[j])break;
                now_num++;
                change_time -= now - nums[j];
            }
            ans = max(ans, now_num);
        }
        return ans;
    }
};

结果：

很慢，可以优化优化。

看题解，可以使用滑动窗口来做。

正确解法：

class Solution {
public:
    int maxFrequency(vector<int>& nums, int k) {
        int n = nums.size();
        sort(nums.begin(), nums.end());
        int ans = 1, l = 0;
        long long total = 0;
        for(int r = 1; r < n; ++r){
            total += (long long)(nums[r] - nums[r - 1]) * (r - l);
            while(total > k){
                total -= nums[r] - nums[l];
                ++l; 
            }
            ans = max(ans, r - l + 1);
        }
        return ans;
    }
};

结果：