LeetCode 2461. 长度为 K 子数组中的最大和

2461. 长度为 K 子数组中的最大和

给你一个整数数组 nums 和一个整数 k 。请你从 nums 中满足下述条件的全部子数组中找出最大子数组和：

• 子数组的长度是 k，且
• 子数组中的所有元素 各不相同 。

返回满足题面要求的最大子数组和。如果不存在子数组满足这些条件，返回 0 。

子数组 是数组中一段连续非空的元素序列。

示例 1：

输入：nums = [1,5,4,2,9,9,9], k = 3
输出：15
解释：nums 中长度为 3 的子数组是：
- [1,5,4] 满足全部条件，和为 10 。
- [5,4,2] 满足全部条件，和为 11 。
- [4,2,9] 满足全部条件，和为 15 。
- [2,9,9] 不满足全部条件，因为元素 9 出现重复。
- [9,9,9] 不满足全部条件，因为元素 9 出现重复。
因为 15 是满足全部条件的所有子数组中的最大子数组和，所以返回 15 。

示例 2：

输入：nums = [4,4,4], k = 3
输出：0
解释：nums 中长度为 3 的子数组是：
- [4,4,4] 不满足全部条件，因为元素 4 出现重复。
因为不存在满足全部条件的子数组，所以返回 0 。

提示：

• 1 <= k <= nums.length <= 10^5
• 1 <= nums[i] <= 10^5

提示 1

Which elements change when moving from the subarray of size k that ends at index i to the subarray of size k that ends at index i + 1?

---

提示 2

Only two elements change, the element at i + 1 is added into the subarray, and the element at i - k + 1 gets removed from the subarray.

---

提示 3

Iterate through each subarray of size k and keep track of the sum of the subarray and the frequency of each element.

 

解法：滑动窗口

class Solution {
    public long maximumSubarraySum(int[] nums, int k) {
        long ans = 0;
        long sum = 0;
        Map<Integer, Integer> window = new HashMap<>();
        for (int i = 0; i < k - 1; i++) {
            sum += nums[i];
            window.merge(nums[i], 1, Integer::sum);
        }
        int i = k - 1;
        while (i < nums.length) {
            sum += nums[i];
            window.merge(nums[i], 1, Integer::sum);
            if (window.size() == k) {
                ans = Math.max(ans, sum);
            }
            int delete = nums[i - k + 1];
            sum -= delete;
            if (window.merge(delete, -1, Integer::sum) == 0) {
                window.remove(delete);
            }
            i++;
        }
        return ans;
    }
}

复杂度分析

• 时间复杂度：O(n)，n 是 数组 nums 的长度。
• 空间复杂度：O(k)。