leetcode 1838. 最高频元素的频数

元素的 频数 是该元素在一个数组中出现的次数。

给你一个整数数组 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 次操作后,能够产生将窗口内的元素全部递增至nums[r],遍历右边界,产生若干个满足条件的窗口,找出最大的一个窗口,该窗口的长度就是最大可能频数

代码

class Solution {
    public int maxFrequency(int[] nums, int k) {

        Arrays.sort(nums);
        int l=0,r=0,n=nums.length,sum=0,max=0;
        if(n==1) return 1;
        while (r<n-1)
        {
            r++;
            sum+=(r-l)*(nums[r]-nums[r-1]);
            while (sum>k)
            {
                sum-=nums[r]-nums[l];l++;
            }
            max= Math.max(max,r-l+1);
            
        }
        return max;
        
    }
}

你可能感兴趣的:(leetcode,leetcode,算法,数据结构,java)