Question:
元素的 频数 是该元素在一个数组中出现的次数。
给你一个整数数组 nums 和一个整数 k 。在一步操作中,你可以选择 nums 的一个下标,并将该下标对应元素的值增加 1 。
执行最多 k 次操作后,返回数组中最高频元素的 最大可能频数 。
Example:
输入:nums = [1,2,4], k = 5
输出:3
解释:对第一个元素执行 3 次递增操作,对第二个元素执 2 次递增操作,此时 nums = [4,4,4] 。
4 是数组中最高频元素,频数是 3 。
Resolution:
本题为滑动窗口的最后一题,根据提示可以得出该最高频元素一定是数组中原来就有的;且我们应该从靠近目标数的元素开始执行操作。
我们列出从start到end索引范围内的元素执行操作后需要的操作次数,根据滑动窗口的特性,求出来end+1后操作次数的变化,并求出来start+1后操作次数的变化,可以发现end+1后,操作次数增加,start+1后操作次数减小。
根据以上分析出来的结果,我们定义start,end指针作为滑动窗口边界,每次循环end+1扩大操作系统并增加需要的操作次数,当滑动窗口内元素需要的操作次数过大超过限定操作次数的时候,我们前移左指针start缩短滑动窗口,直到窗口内的元素需要的操作次数小于限定的操作次数,每次循环更新最大可能频数。
Code:
public int maxFrequency(int[] nums, int k) {
Arrays.sort(nums);
int n = nums.length, start = 0, end = 1, res = 1;
long total = 0;
for (; end < n; end++) {
total += (long) (nums[end] - nums[end - 1]) * (end - start);
while (total > k) {
total -= nums[end] - nums[start];
++start;
}
res = Math.max(res, end - start + 1);
}
return res;
}