题目链接
题目:
给你一个整数数组 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
官答二:
class Solution {
vector<int> cnt;
public:
int numberOfSubarrays(vector<int>& nums, int k) {
int n = (int)nums.size();
cnt.resize(n + 1, 0);
int odd = 0, ans = 0;
cnt[0] = 1;
for (int i = 0; i < n; ++i) {
odd += nums[i] & 1;
ans += odd >= k ? cnt[odd - k] : 0;
cnt[odd] += 1;
}
return ans;
}
};
分析:
官方题解下cnt数组一直没搞明白到底是怎么个记法,后来一步步调试才大致看懂。cnt数组的第i位元素a,表示的是父数组里以第i个奇数做优美子数组里首个奇数的优美子数组数量(且子数组里最后一位不是偶数时,这个情况要单独说明)。
以以下示例举例:
输入:nums = [2,2,2,1,2,1,2,1,2], k = 2
输出:12
在开头遍历过程中cnt[0]不断增大,直到变成4才不再增加,再来看一下包含nums[3]这个奇数的有几个数组:
(1)[2,2,2,1,2,1]
(2)[2,2,1,2,1]
(3)[2,1,2,1]
(4)[1,2,1]
1.而当遍历到nums[4]时,ans=ans+cnt[0]=ans+4。
这里比较好理解,就是前面的这4个数组末尾加了个2;而此时cnt[0]不再变化,而是cnt[1]++,因为这个2相当于是第2个奇数做最优子数组开头奇数的情况了,就如同前面cnt[0]增大的情况,这里也是这么计算。因为第2个奇数和第3个奇数之间只有1个2,因此cnt[1]只增到2,ans的增加也是改为了ans=ans+cnt[1]=ans+2。
2.遍历到了到了第3个奇数,此时cnt[1]停止增加,cnt[2]开始增加,但在这个示例里其实跟它关系不大,为什么呢?因为并没有以第3个奇数做开头的优美子数组,cnt[2]的增加只是在为后续还可能出现的奇数做准备。只要不出现第4个奇数,cnt[2]就永远用不到。
真的想了很久,思路感觉和官方题解不一样,主要是官方题解着实看不懂啊。。。