1838. 最高频元素的频数
1838. 最高频元素的频数 (Medium)
元素的 频数 是该元素在一个数组中出现的次数。
给你一个整数数组 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⁵1 <= nums[i] <= 10⁵1 <= k <= 10⁵
分析
先排序,然后用滑动窗口来确定结果,对于窗口 [lo, hi], nums[hi] 是最大值,将整个窗口调整成 nums[hi] 后,和为 nums[hi]*(hi-lo+1), 如果该值比实际 sum 加上 k 还大,需要缩小窗口左边界。
func maxFrequency(nums []int, k int) int {
sort.Ints(nums)
lo := 0
sum := 0
res := 0
for hi, v := range nums {
sum += v
for sum+k < v*(hi-lo+1) {
sum -= nums[lo]
lo++
}
res = max(res, hi-lo+1)
}
return res
}