【l前端算法】1248. 统计「优美子数组」,动态规划

623 阅读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

解题代码:
var numberOfSubarrays = function(nums, k) {
 let odd = [-1];
 for(var i=0;i<nums.length;++i){
     if(nums[i]%2==1){
  odd.push(i)
     }
 }
 odd.push(nums.length);
 let result = 0;
 for(var j=1;j+k<=odd.length-1;++j){
   result += (odd[j] - odd[j - 1])*(odd[j + k] - odd[j + k - 1])
 }
 return result
} 

本文使用 mdnice 排版