1838. 最高频元素的频数-每日一题

一、题目

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

  给你一个整数数组 nums 和一个整数 k 。在一步操作中，你可以选择 nums 的一个下标，并将该下标对应元素的值增加 1 。

  执行最多 k 次操作后，返回数组中最高频元素的 最大可能频数 。

点击查看原题
难度级别：中等

二、思路

1）排序+暴力搜索

  由于对数字只进行增加1的操作，那么选定一个nums[i]后，只有对它小的数字进行操作才会增加其数量，所以先进行排序，这样比nums[i]小的元素都在其左侧，又由于每次只能增加1，那么根据贪心思想，肯定优先选择最大且小于nums[i]的元素
  通过一次遍历，每个元素都向左搜索，找到最多能增加到nums[i]值的元素数量

2）排序+滑动窗口

  根据上个方法的思想，其实不需要每次都对nums[i]左边重新进行搜索，计算最多能又多少个增加到nums[i]值的元素数量时，就已经统计了x个元素增加到nums[i]值的需要消耗多少次增加1，那么增加到nums[i+1]就是再消耗掉x*(nums[i+1]-nums[i])即可，设定两个左右指针，如果增加1的次数不够消耗，就左指针右移，将该位置消耗的增加1的次数返还即可

三、代码

1）排序+暴力搜索

class Solution {
    public int maxFrequency(int[] nums, int k) {
        int ans = 0;
        Arrays.sort(nums);
        for (int i = nums.length - 1; i >= 0; i--) {
            if (i != nums.length -1 && nums[i] == nums[i+1]) {
                continue;
            }
            int tempRecord = 1;	// 记录数量
            int loc = i - 1;
            int consume = k;
            while (consume > 0 && loc >= 0) {	// 无可以增加的数量，或者搜索到最左边就退出
                int margin = nums[i] - nums[loc];
                consume -= margin;
                if (consume >= 0) {
                    tempRecord++;
                }
                loc--;
            }
            ans = Math.max(ans, tempRecord);
        }
        return ans;
    }
}

  时间复杂度为O(nlogn+kn)，空间复杂度为O(logn)

2）排序+滑动窗口

class Solution {
    public int maxFrequency(int[] nums, int k) {
        int ans = 1;
        int left = 0, consume = 0;
        Arrays.sort(nums);
        for (int right = 1; right < nums.length; right++) {
            consume += (right - left) * (nums[right] - nums[right - 1]);
            if (consume > k) {
                consume -= (nums[right] - nums[left]);
                left++;
            }
            ans = Math.max(ans, right - left + 1);
        }
        return ans;
    }
}

  时间复杂度为O(nlogn)，空间复杂度为O(logn)