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