给你一个整数数组 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 <= 105
1 <= nums[i] <= 105
滑动窗口
例如:
class Solution { public: long long maximumSubarraySum(vector<int>& nums, int k) { long long res=0,sum=0; // 定义结果变量 res 和子数组和 sum,初始化为 0 int count=0; // 记录子数组中不同元素的个数 int visit[100001]={0}; // 使用数组 visit 记录每个元素出现的次数,数组大小为 100001,初始化为 0 for (int i = 0; i <k ; ++i) { // 遍历数组 nums 的前 k 个元素 sum+=nums[i]; // 更新子数组和 visit[nums[i]]++; // 更新元素 nums[i] 的出现次数 if (visit[nums[i]]==1){ // 如果元素 nums[i] 是第一次出现 count++; // 增加不同元素的计数 } } if (count==k){ // 如果子数组中的不同元素个数等于 k res=sum; // 更新结果为当前子数组和 } for (int i = k; i <nums.size() ; ++i) { // 从第 k 个元素开始遍历数组 nums sum=sum+nums[i]-nums[i-k]; // 更新子数组和,减去最旧的元素,加上最新的元素 visit[nums[i]]++; // 更新最新元素的出现次数 if (visit[nums[i]]==1){ // 如果最新元素是第一次出现 count++; // 增加不同元素的计数 } visit[nums[i-k]]--; // 更新最旧元素的出现次数 if (visit[nums[i-k]]==0){ // 如果最旧元素不再出现 count--; // 减少不同元素的计数 } if (count==k){ // 如果子数组中的不同元素个数等于 k res= max(res,sum); // 更新结果为当前子数组和与历史最大值的较大者 } } return res; // 返回结果 } };