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

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

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

子数组的长度是 k，且
子数组中的所有元素 各不相同 。
返回满足题面要求的最大子数组和。如果不存在子数组满足这些条件，返回 0 。

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

• 输入：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 。

题目链接

前缀和（TLE）

先算出前缀和，然后遍历每一个子区间长度并记录合法的最大值

class Solution {
public:
    long long maximumSubarraySum(vector<int>& nums, int k) {
        long long int res=0;
        vector<long long>presum(nums.size()+1);
        presum[0]=0;
        for(int i=1;i<=nums.size();i++) presum[i]=presum[i-1]+nums[i-1];
        for(int i=0;i+k<=nums.size();i++){
            long long int sum=presum[i+k]-presum[i];
            //cout<<sum<<endl;
            map<int,int>book;
            int flag=0;
            for(int j=i;j<i+k;j++){
                if(book[nums[j]]){
                    flag=1;
                    break;
                }
                book[nums[j]]=1;
            }
            if(!flag){
                res=max(res,sum);
            }
        }
        return res;
    }
};

滑动窗口

class Solution {
public:
    long long maximumSubarraySum(vector<int>& nums, int k) {
        long long int res=0;
        long long int sum=0;
        map<int,int>book;
        int left=0,right=0;
        for(right;right<k;right++){
            book[nums[right]]++;
            sum+=nums[right];
        }
        if(book.size()==k)  res=sum;
        for(right;right<nums.size();right++){
            book[nums[right]]++;
            sum+=nums[right];
            book[nums[left]]--;
            if(book[nums[left]]==0) book.erase(nums[left]);
            sum-=nums[left];
            left++;
            if(book.size()==k) res=max(res,sum);
        }
        return res;
    }
};