LeetCode 1248. 统计「优美子数组」(前缀)+(滑窗)

1.题目

给你一个整数数组 nums 和一个整数 k。
如果某个 连续 子数组中恰好有 k 个奇数数字，我们就认为这个子数组是「优美子数组」。
请返回这个数组中「优美子数组」的数目。

示例 1：

输入：nums = [1,1,2,1,1], k = 3
输出：2
解释：包含 3 个奇数的子数组是 [1,1,2,1] 和 [1,2,1,1] 。
示例 2：

输入：nums = [2,4,6], k = 1
输出：0
解释：数列中不包含任何奇数，所以不存在优美子数组。
示例 3：

输入：nums = [2,2,2,1,2,2,1,2,2,2], k = 2
输出：16
 

提示：

1 <= nums.length <= 50000
1 <= nums[i] <= 10^5
1 <= k <= nums.length

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-number-of-nice-subarrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.题解

前缀：

这道题目可以等价变换为：奇数视为1，偶数视为0，统计有多少个子数组和为k（包含了k个1）
用v[i]来表示数组前i项中，奇数的个数，则改题可以转化为求得下标i到j中奇数个数为k的i，j对。
pre[i]-pre[j-1]=k;
pre[i]-k=pre[j-1];
pre[j]可以用pre变量代替,v来记录pre[i] 出现的次数，从左往右边更新v边计算答案，那么以 i 结尾的答案v[pre-k] 即可 O(1) 得到。最后的答案即为所有下标结尾的「优美子数组」个数之和。
数组从左往右开始遍历，统计奇数的个数，满足条件用res进行记录。

class Solution {
public:
    int numberOfSubarrays(vector<int>& nums, int k) {
        vector<int> v(nums.size()+1);
        int pre = 0;			    //记录奇数的个数
        v[0] = 1;
        int res = 0;
        for(auto x:nums){
            pre += x & 1;
            v[pre]++;				//记录pre[i] 出现的次数
            if(pre >= k) res += v[pre-k];
        }
        return res;
    }
};

双指针（滑窗）：

这题的难点在于边界，要自己设置边界。

class Solution {
public:
	int numberOfSubarrays(vector<int>& nums, int k) {
		vector<int>odd_index = { -1 };//奇数索引，注意边界
		for (int i = 0; i < nums.size(); i++)
		{
			if (nums[i] & 1)
			{
				odd_index.push_back(i);
			}
		}
		odd_index.push_back(nums.size());//注意边界
		int res = 0;
		for (int i = 1; i + k - 1 <= odd_index.size() - 2; i++)//基本区间为[i,i+k-1]
		{
			//可以向左边扩展（而不遇到新的奇数）的最大距离
			int left = odd_index[i] - odd_index[i - 1];
			//可以向右边扩展（而不遇到新的奇数）的最大距离
			int right = odd_index[i + k] - odd_index[i + k - 1];
			res += left*right;
		}
		return res;

	}
};

原文：https://blog.csdn.net/ShenHang_/article/details/105653724