天天看點

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 {
public:
    int maxFrequency(vector<int>& nums, int k) {
        sort(nums.begin(), nums.end());

        int ans = 0;
        long long sum = 0;
        for (int l = 0, r = 0; r < nums.size(); r++) {
            while ((long long)nums[r] * (r - l) - sum > k) {
                sum -= nums[l++];
            }
            sum += nums[r];
            ans = max(ans, r - l + 1);
        }

        return ans;
    }
};