题目:
元素的 频数 是该元素在一个数组中出现的次数。
给你一个整数数组 nums 和一个整数 k 。在一步操作中,你可以选择 nums 的一个下标,并将该下标对应元素的值增加 1 。执行最多 k 次操作后,返回数组中最高频元素的 最大可能频数 。
思路:
1.首先对数组进行排序
2.利用滑动窗口,窗口左边界为i,右边界为j,看当前窗口的值通过增加k后,能够使所有数都增大到nums[j]。即是否满足:k + nums[i]+…+nums[j] >= nums[j] * (j - i + 1) 。
如果可以,则移动右边界,扩大窗口;
如果不可以,则移动左边界,滑动窗口。
解答:
class Solution:
def maxFrequency(self, nums: List[int], k: int) -> int:
nums.sort()
i,j,total=0,0,k
#问题转化:排序后 nums[j] * (j - i + 1) <= k + nums[i]+...+nums[j]
for j,num in enumerate(nums):
total+=num
#如果不符合条件,则窗口向右滑动
if total<num*(j-i+1):
total-=nums[i]
i+=1
#否则窗口扩张
return j-i+1