题目描述:
给你一个整数数组 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
方法1:
主要思路:解题汇总
(1)使用哈希函数,统计奇数的个数的各个情形;
class Solution {
public:
int numberOfSubarrays(vector<int>& nums, int k) {
if(nums.size()==0){//特殊的情形
return 0;
}
int n=nums.size();
unordered_map<int,int> mp;
int cur_count=0;//数组从索引1到当前位置的所有的奇数的个数
int res=0;//统计满足要求情形的个数
for(int i=0;i<n;++i){//遍历数组
if(nums[i]&1){//若当前数字是奇数
++cur_count;
}
//若当前奇数的个数为k则对统计的结果加1
if(cur_count==k){
++res;
}
++mp[cur_count];//当前奇数个数可能出现的情形
if(mp.count(cur_count-k)){//若当前奇数的个数减去k,获得值在之前统计过,说明这些位置之间也是满足要求,加到统计结果
res+=mp[cur_count-k];
}
}
return res;
}
};
方法2:
主要思路:
(1)思路和上述一致,只不过将哈希使用数组进行替代;
class Solution {
public:
int numberOfSubarrays(vector<int>& nums, int k) {
if(nums.size()==0){
return 0;
}
int n=nums.size();
vector<int> mp(n+1,0);//统计可能出现的计数个数的各个情形
mp[0]=1;//初始化,处理奇数个数第一次为 k的情形
int cur_count=0;//统计从0到当前位置,奇数的个数
int res=0;//统计满足要求的情形
for(int i=0;i<n;++i){
if(nums[i]&1){//统计奇数的个数
++cur_count;
}
++mp[cur_count];//统计改个数可能的情形
if(cur_count>=k){//说明之前出现过满足要求的结果,进行统计
res+=mp[cur_count-k];
}
}
return res;
}
};