题目来源:https://leetcode-cn.com/problems/frequency-of-the-most-frequent-element/
大致题意:
给定一个数组nums(元素值在1~10^5)和操作数k。假定现在每次操作可以使数组某个元素加 1,求出在不多于k次操作的情况下,数组中可能出现的最高频次的数的频次。
思路
k次操作最多给一个数 +k,那么目标数和原数的差应不大于k。
排序加滑动窗口
先将数组排序。
使用一个窗口在数组上滑动,窗口中的数可以经过小于k次操作变成右边界的数,且所有操作次数的和也不大于k。于是在滑动的过程中,窗口的最大长度即为所求的最高频次。
- 初始化时,窗口左边界 l 为 0,使用一个 sum 存下操作次数之和,初始为 0
- 从索引 i = 1 开始遍历数组,每次将当前数与上一个元素相减,再乘上 i - l(即之前窗口所有数都变成当前元素需要再加上的操作次数),然后将新加的操作次数加入sum
- 若 sum > k,则将 sum 减去当前元素与左边界的差,即窗口左边界右移一步。重复这个过程,直至 sum <= k
- 对比 之前的窗口最大长度 maxF 与 当前窗口长度 i - l + 1,取最大值
代码:
public int maxFrequency(int[] nums, int k) {
Arrays.sort(nums);
int maxF = 1;
int l = 0;
int sum = 0;
for (int i = 1; i < nums.length; i++) {
// 将之前窗口的数变为当前元素需要加上的操作次数
sum += (nums[i] - nums[i-1]) * (i - l);
// 更新窗口大小
while (sum > k) {
sum -= nums[i] - nums[l++];
}
// 获取最大窗口长度
maxF = Math.max(maxF, i - l + 1);
}
return maxF;
}