【周赛318 LeetCode 6230】长度为 K 子数组中的最大和

题目描述

给你一个整数数组 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 <= 10^5
    1 <= nums[i] <= 10^5

思路

1.暴力+hash 超时

利用set去判断当前的元素是否已经计算过，也就是是否包含重复元素。有则跳过结束本次循环，否则进行加入。

long long maximumSubarraySum(vector<int>& nums, int k) {
	long long int ans=0;
	for(int i=0;i<nums.size()-k+1;i++){
		unordered_set<int> temp;
		long long int sum=0;
		int flag=1;
		for(int j=i;j<i+k;j++){
			//判断是否存在
			if(temp.find(nums[j])==temp.end()){
				temp.insert(nums[j]);
				sum+=nums[j];				
			}
			else{//已经存在则 结束本次循环 并添加标志位表示这次的sum不参与比较
				flag=0;
				break;
			}
		}
		if(flag==1 && sum>ans)
			ans=sum;
	}
	return ans;
}

2.滑动窗口 AC

第一个思路超时了，后面想到了滑动窗口的思想。于是在之前的基础上，利用**双指针(快慢指针)**实现滑动窗口。

long long maximumSubarraySum(vector<int>& nums, int k) {
	long long int ans=0,sum=0;
	int len=nums.size();
	set<int> temp;//set是有序的，不能使用unordered_set 方便删除
	int left=0,right=0;
	while(right-left<k && right<len){//当前right-left表示当前的滑动窗口大小
		if(temp.find(nums[right])!=temp.end()){//若存在，说明有元素重复，则清空
			temp.clear();//清空当前的集合
			left=right;
			sum=0;
		}
		sum+=nums[right];
		temp.insert(nums[right]);
		if(temp.size() == k){//当前滑动窗口达到最大值，于是进行比较
			if(sum>ans)
				ans=sum;
			sum-=nums[left];
			temp.erase(nums[left]);//删除滑动窗口的左边起始值。
			left++;//滑动窗口移动
		}
		right++;//当前滑动窗口扩大
	}
	return ans;
}