元素的 频数 是该元素在一个数组中出现的次数。
给你一个整数数组 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
int maxFrequency(vector<int> &nums, int k) {
sort(nums.begin(), nums.end()); // 对数组进行排序
long long temp = 0; // 用于临时存储当前窗口内的元素总和
int ans = 0; // 存储最大频数的变量
for (int l = 0, r = 0; r < nums.size(); ++r) { // 使用双指针维护滑动窗口
while ((static_cast<long long>(r - l)) * nums[r] > (k + temp)) { // 如果当前窗口不能通过k次操作使得全部元素相等
temp -= nums[l]; // 减去窗口最左侧的元素
l++; // 左指针右移
}
ans = max(ans, r - l + 1); // 更新最大频数
temp += nums[r]; // 添加窗口右侧的元素到临时总和中
}
return ans; // 返回最大频数
}