力扣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

思路

暴力破解

代码

java 复制代码

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);
            }
            else
                map.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;

    }
}