给你一个整数数组 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
一开始的思路是找到包含k个奇数的子数组,然后计算开头到前一个奇数之间的偶数个数和结尾到下一个奇数之间的偶数个数,最后计算可组成的子数组的个数
class Solution {
public int numberOfSubarrays(int[] nums, int k) {
int len = nums.length;
int ans = 0;
List<Integer> list = new ArrayList<>();//保存奇数的索引
for(int i=0;i<len;i++){
if((nums[i]&1)==1){
list.add(i);
}
}
if(list.size()<k) return 0;
for(int j=0;j+k<list.size()+1;j++){
int head = 0,tail = 0;//头尾偶数的个数
if(j>0){
head = list.get(j)-list.get(j-1)-1;
}else{
head=list.get(j);
}
if(j+k<list.size()){
tail=list.get(j+k)-list.get(j+k-1)-1;
}else{
tail=len-list.get(j+k-1)-1;
}
for(int m=0;m<=head;m++){
for(int n=0;n<=tail;n++){
ans++;
}
}
}
return ans;
}
}
根据官方题解做优化
class Solution {
public int numberOfSubarrays(int[] nums, int k) {
int len = nums.length;
int ans = 0;
List<Integer> list = new ArrayList<>();
list.add(-1);
for(int i=0;i<len;i++){
if((nums[i]&1)==1){
list.add(i);
}
}
list.add(len);
if(list.size()<k) return 0;
for(int j=1;j+k<list.size();j++){
ans+=(list.get(j)-list.get(j-1))*(list.get(j+k)-list.get(j+k-1));
}
return ans;
}
}
进一步优化
class Solution {
public int numberOfSubarrays(int[] nums, int k) {
int len = nums.length;
int odd=0, ans = 0;
int[] cnt = new int[len+1];
cnt[0] = 1;
for(int i=0;i<len;i++){
odd+=nums[i]&1;
ans+=odd>=k?cnt[odd-k]:0;
cnt[odd]++;
}
return ans;
}
}