leetcode 2841. 几乎唯一子数组的最大和 2461. 长度为 K 子数组中的最大和

两道题本质是一样的，所以放一起了

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

使用unordered_map；unordered_multiset可能也可以，但是不如前者方便

class Solution {
public:
    long long maxSum(vector<int>& nums, int m, int k) {
        int size = nums.size();
        long long now = 0;
        unordered_map<int,int> numAdded;
        for(int i = 0;i < k;i++){
            now += nums[i];
            numAdded[nums[i]]++;//如果原本没有这个键，那么就会添加，对应的值自增1
        }
        long long resMax = 0;//题目说了，如果不存在几乎唯一子数组，请返回0
        if(numAdded.size() >= m)  resMax = now;  

        for(int i = k;i < size;i++){
            if(numAdded[nums[i-k]] == 1)  numAdded.erase(nums[i-k]);//如果将要移出的元素在原本的unordered_map只剩下一个，直接从unordered_map中删除
            else  numAdded[nums[i-k]]--;
            now = now - nums[i-k] + nums[i];
            numAdded[nums[i]]++;
            if(numAdded.size() >= m && now > resMax)  resMax = now;
        }
        
        return resMax;
    }
};

 

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

class Solution {
public:
    long long maximumSubarraySum(vector<int>& nums, int k) {
        int size = nums.size();
        long long now = 0, maxRes = 0;
        unordered_map<int,int> numAdded;
        for(int i = 0;i < k;i++){
            now += nums[i];
            numAdded[nums[i]]++;
        }
        if(numAdded.size() == k)  maxRes = now;

        for(int i = k;i < size;i++){
            if(numAdded[nums[i-k]] == 1)  numAdded.erase(nums[i-k]);
            else  numAdded[nums[i-k]]--;
            //这里很容易忘记执行numAdded[nums[i-k]]--;或者要 numAdded[nums[i-k]] != 1 才能执行numAdded[nums[i-k]]--;
            numAdded[nums[i]]++;
            now = now - nums[i-k] + nums[i];
            if(numAdded.size() == k && maxRes < now)  maxRes = now;
        }
        return maxRes;
    }
};