给你一个整数数组 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; // 返回结果
}
};