元素的 频数 是该元素在一个数组中出现的次数。
给你一个整数数组 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;
}
};