给你一个整数数组 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
思路:
对于一个以奇数a开头,奇数b结尾的优美子数组,在这个数组的前后加上若干个偶数后的数组仍然符合条件。假设奇数a前有x个偶数,奇数b后有y个偶数,那么就可以得到(x+1)*(y+1)个优美子数组。所以只需要统计奇数之间偶数的个数和奇数所在的位置就可以得到答案。遍历两遍数组,时间复杂度为O(n)。
#include <iostream>
#include <vector>
#include <map>
using namespace std;
class Solution {
public:
int numberOfSubarrays(vector<int>& nums, int k) {
int res=0;
int index=0;
int numOfEven=1;
int len=(int)nums.size();
vector<int> cnt;
vector<int> pos;
for(int i=0;i<len;i++){
if(nums[i]%2==0) numOfEven++;
else{
cnt.push_back(numOfEven);
index++;
numOfEven=1;
cnt.push_back(nums[i]);
pos.push_back(index);
index++;
}
}
cnt.push_back(numOfEven);
if(pos.size()<k) return 0;
int start,end;
start=0;
end=start+k-1;
while(end<pos.size()){
res += (cnt[pos[start]-1] * cnt[pos[end]+1]);
start++;
end++;
}
return res;
}
};
int main(int argc, const char * argv[]) {
vector<int> input{2044,96397,50143};
Solution s;
cout<<s.numberOfSubarrays(input, 1);
return 0;
}