2841.几乎唯一子数组的最大和

题目

给你一个整数数组 nums 和两个正整数 m 和 k 。

请你返回 nums 中长度为 k 的 几乎唯一 子数组的 最大和 ，如果不存在几乎唯一子数组，请你返回 0 。

如果 nums 的一个子数组有至少 m 个互不相同的元素，我们称它是 几乎唯一 子数组。

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

示例 1：

1
2
3

输入：nums = [2,6,7,3,1,7], m = 3, k = 4
输出：18
解释：总共有 3 个长度为 k = 4 的几乎唯一子数组。分别为 [2, 6, 7, 3] ，[6, 7, 3, 1] 和 [7, 3, 1, 7] 。这些子数组中，和最大的是 [2, 6, 7, 3] ，和为 18 。

示例 2：

1
2
3

输入：nums = [5,9,9,2,4,5,4], m = 1, k = 3
输出：23
解释：总共有 5 个长度为 k = 3 的几乎唯一子数组。分别为 [5, 9, 9] ，[9, 9, 2] ，[9, 2, 4] ，[2, 4, 5] 和 [4, 5, 4] 。这些子数组中，和最大的是 [5, 9, 9] ，和为 23 。

示例 3：

1
2
3

输入：nums = [1,2,1,2,1,2,1], m = 3, k = 3
输出：0
解释：输入数组中不存在长度为 k = 3 的子数组含有至少  m = 3 个互不相同元素的子数组。所以不存在几乎唯一子数组，最大和为 0 。

提示：

1 <= nums.length <= 2 * 104
1 <= m <= k <= nums.length
1 <= nums[i] <= 109

题解

class Solution {
    public long maxSum(List<Integer> nums, int m, int k) {
        HashMap<Integer ,Integer> map = new HashMap<>();
        int n=nums.size();
        long max=0;
        long sum=0; // 注意要使用long型，否则越界
        for(int i=0;i<n;i++){
            sum+=nums.get(i);
            if(map.containsKey(nums.get(i))){
                map.put(nums.get(i),map.get(nums.get(i))+1);
            }else{
                map.put(nums.get(i),1);
            }
            if(i<k-1){
                continue;
            }
            // System.out.printlni"map"+map.size());
            if(sum>max&&map.size()>=m){
                max=sum;
            }
            sum-=nums.get(i-(k-1));
            if(map.get(nums.get(i-(k-1)))==1){
                map.remove(nums.get(i-(k-1)));
            }else{
                map.put(nums.get(i-(k-1)),map.get(nums.get(i-(k-1)))-1);
            }
            //map.remove(i-(k-1));
        }
        return max;
    }
}