元素的频数是该元素在一个数组中出现的次数。
给你一个整数数组 nums 和一个整数 k 。在一步操作中,你可以选择 nums 的一个下标,并将该下标对应元素的值增加 1 。
执行最多 k 次操作后,返回数组中最高频元素的 最大可能频数 。
示例 1:
输出:3
解释:对第一个元素执行 3 次递增操作,对第二个元素执 2 次递增操作,此时 nums = [4,4,4] 。
4 是数组中最高频元素,频数是 3 。
示例 2:
输出: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^51 <= nums[i] <= 10^51 <= k <= 10^5
思考过程: 遍历所有情况再优化可以解决所有的算法问题。
第一步:将 nums 数组从小到大排序。
第二步:用变量 i 从前往后依次遍历,把每一次取到的 nums[i] 值看成完成递增操作后的最高频元素。
第三步:用变量 j 从 变量 i 开始从后往前依次遍历,检查 num[j] 是否能通过递增操作变成最高频元素。
时间复杂度:O(n^2)
空间复杂度:O(1)
优化思路: 由于这题的数据量较小,在采用HashMap进行 nums 数组的频数统计,从而达到去重的效果后即可完成提交。
时间复杂度:O(n^2)
空间复杂度:O(1)
Ps:太久没刷题,刚开始复健,自己总结一下发现我这优化了个寂寞。
代码:
class Solution {
public int maxFrequency(int[] nums, int k) {
//HashMap计数
HashMap<Integer, Integer> map = new HashMap<>();
for (int i = 0; i < nums.length; i++) {
map.put(nums[i], map.getOrDefault(nums[i], 0) + 1);
}
//按数字大小排序
int index = 0;
int[][] fre = new int[map.size()][2];
for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
fre[index][0] = entry.getKey();
fre[index][1] = entry.getValue();
index++;
}
Arrays.sort(fre,(a, b) -> (a[0] - b[0]));
//扫描
int maxFre = 0;
for (int i = 0; i < fre.length; i++) {
int this_i_max = 0;
int this_i_k = k;
for (int j = i; j >= 0; j--) {
int delta = fre[i][0] - fre[j][0];
if (delta == 0) {
this_i_max += fre[j][1];
} else if ((this_i_k - delta * fre[j][1]) >= 0) {
this_i_k -= delta * fre[j][1];
this_i_max += fre[j][1];
} else {
this_i_max += this_i_k / delta; break;
}
}
maxFre = Math.max(maxFre, this_i_max);
}
return maxFre;
}
}