LeetCode 5248:统计最美子数组

题目描述:

给你一个整数数组 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

解释:数列中不包含任何奇数,所以不存在优美子数组。

也就是说,我们需要从数组中统计出具有k个奇数数字的子数组。这里子数组是连续的。那么我们用一个map来记录原数组从最左侧开始的具有a个奇数的数组的数目。

比如我们给定数组:nums=[1,1,2,1,1,]。则map[0]=1,(表示为空),map[1]=1,map[2]=1,map[3]=2,map[4]=1。

则我们下一步需要求的为子数组中含有k个奇数的子数组数目。则若当前map[s]=a,map[s-k]=b,则此时对应的含有k个奇数的子数组的数目为

a-b。按照这个想法很容易的就可以写出代码了:


AC代码

你可能感兴趣的:(LeetCode 5248:统计最美子数组)