力扣3318——计算子数组的 x-sum I(偷懒版)

题目

给你一个由 n 个整数组成的数组 nums，以及两个整数 k 和 x。

数组的 x-sum 计算按照以下步骤进行：

• 统计数组中所有元素的出现次数。
• 仅保留出现次数最多的前 x 个元素的每次出现。如果两个元素的出现次数相同，则数值 较大
  的元素被认为出现次数更多。
• 计算结果数组的和。
  注意，如果数组中的不同元素少于 x 个，则其 x-sum 是数组的元素总和。
  返回一个长度为 n - k + 1 的整数数组 answer，其中 answer[i] 是 子数组 nums[i…i + k - 1] 的 x-sum。
  子数组 是数组内的一个连续 非空 的元素序列。

示例

示例 1：

输入：nums = [1,1,2,2,3,4,2,3], k = 6, x = 2
输出：[6,10,12]
解释：
对于子数组 [1, 1, 2, 2, 3, 4]，只保留元素 1 和 2。因此，answer[0] = 1 + 1 + 2 + 2。
对于子数组 [1, 2, 2, 3, 4, 2]，只保留元素 2 和 4。因此，answer[1] = 2 + 2 + 2 + 4。注意 4 被保留是因为其数值大于出现其他出现次数相同的元素（3 和 1）。
对于子数组 [2, 2, 3, 4, 2, 3]，只保留元素 2 和 3。因此，answer[2] = 2 + 2 + 2 + 3 + 3。

示例 2：

输入：nums = [3,8,7,8,7,5], k = 2, x = 2
输出：[11,15,15,15,12]
解释：
由于 k == x，answer[i] 等于子数组 nums[i…i + k - 1] 的总和。

提示

• 1 <= n == nums.length <= 50
• 1 <= nums[i] <= 50
• 1 <= x <= k <= nums.length

思路

暴力破解

代码

class Solution {
public static int[] findXSum(int[] nums, int k, int x) {int []answer = new int[nums.length - k + 1];for(int i=0;i<answer.length;i++){answer[i] = findSum(nums,i,k,x);}return answer;}public static int findSum(int[] nums, int start, int k, int x) {Map<Integer, Integer> map = new HashMap<>();for (int i = start; i < start + k; i++) {if(map.containsKey(nums[i])){map.put(nums[i],map.get(nums[i])+1);}elsemap.put(nums[i],1);}LinkedHashMap<Integer, Integer> sortedMap = map.entrySet().stream().sorted((Comparator<? super Map.Entry<Integer, Integer>>) Map.Entry.<Integer, Integer>comparingByValue(Comparator.reverseOrder()).thenComparing(Map.Entry.comparingByKey(Comparator.reverseOrder()))).collect(Collectors.toMap(Map.Entry::getKey,Map.Entry::getValue,(e1, e2) -> e1,LinkedHashMap::new));int sum = 0;int index = 0;for(var entry : sortedMap.entrySet()){if(index >= x)break;sum += entry.getKey()*entry.getValue();index++;}return sum;}
}