题目描述
题干:
元素的 频数 是该元素在一个数组中出现的次数。
给你一个整数数组 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
题解思路
返回操作之后出现次数最多元素的次数,题解上给了前缀和、二分查找和滑动创建口好几种相关标签
其实感觉只用了了滑动窗口的思想吧,因为每个元素都有可能成为操作后出现次数最多的元素
所以我们首先对数组进行排序,然后用模拟一个窗口从左往右移动,每次都按照最右侧元素进行补充
如果需要操作的数大于k,就将左侧边框变窄,左侧边框右移一位,因为他需要的数最多
每次右移一个窗口之后统计窗口的宽度去最大值,即操作后出现次数最最多元所出现的次数
正确代码
public int maxFrequency(int[] nums, int k) {
Arrays.sort(nums);
int length = nums.length;
long total = 0;
int x = 0, res = 1;
for (int i = 1; i < length; i++) {
total += (long)(nums[i] - nums[i - 1]) * (i - x);
while (total > k) {
total -= nums[i] - nums[x];
x++;
}
res = Math.max(res, i - x + 1);
}
return res;
}
总结
滑动窗口的思想类用于存在左右边界并且对中间点元素有限制的情况,类似于双指针
如果文章存在问题或者有更好的题解,欢迎在评论区斧正和评论,各自努力,你我最高处见