【C++滑动窗口】1248. 统计「优美子数组」|1623

本文涉及的基础知识点

C++算法：滑动窗口及双指针总结

LeetCode1248. 统计「优美子数组」

给你一个整数数组 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⁵
1 <= k <= nums.length

滑动窗口

nums[i…j1]包括k个奇数的数字，且j1最小。
nums[i…j2]包括k+1个奇数的数字，且j2最小。
枚举i，ans += j2-j1。此方法称为三指针。
为了简化代码，可以用两个滑动窗口。
一，N-j1，以i开头，包括k个及以上奇数的数目。
二，N-j2，以i开头，包括k+1个及以上奇数的数目。

代码

核心代码

class Solution {public:int numberOfSubarrays(vector<int>& nums, int k) {const int N = nums.size();auto Do = [&](int len) {int cnt = 0;long long res = 0;for (int i = 0, j = 0; i < N; i++) {while ((j < N) && (cnt < len )) {cnt += bool(nums[j] & 1); j++;}if (cnt >= len){res += N - j+1;}cnt -= bool(nums[i] & 1);}return res;};return Do(k) - Do(k + 1);}};

单元测试

vector<int> nums; int k;TEST_METHOD(TestMethod11){nums = { 1,1,2,1,1 },k=3;auto res = Solution().numberOfSubarrays(nums, k);AssertEx(2, res);}TEST_METHOD(TestMethod12){nums = { 2,4,6 }, k = 1;auto res = Solution().numberOfSubarrays(nums, k);AssertEx(0, res);}TEST_METHOD(TestMethod13){nums = { 2,2,2,1,2,2,1,2,2,2 }, k = 2;auto res = Solution().numberOfSubarrays(nums, k);AssertEx(16, res);}

扩展阅读

我想对大家说的话
工作中遇到的问题，可以按类别查阅鄙人的算法文章，请点击《算法与数据汇总》。
学习算法：按章节学习《喜缺全书算法册》，大量的题目和测试用例，打包下载。重视操作
有效学习：明确的目标及时的反馈拉伸区（难度合适）专注
闻缺陷则喜(喜缺)是一个美好的愿望，早发现问题，早修改问题，给老板节约钱。
子墨子言之：事无终始，无务多业。也就是我们常说的专业的人做专业的事。
如果程序是一条龙，那算法就是他的是睛
失败+反思=成功成功+反思=成功