【LeetCode】1248. 统计「优美子数组」

2021-02-09 19:18

阅读:391

标签:c++   复杂   一个   示例   连续   +=   数列   ber   arrays   

题目

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

思路一:数学

代码

时间复杂度:O(n)
空间复杂度:O(n)

class Solution {
public:
    int numberOfSubarrays(vector& nums, int k) {
        int size = nums.size(), cnt = 0, res = 0;
        vector odd(size + 2);
        odd[0] = -1;
        for (int i = 0; i 

思路二:前缀和

代码

时间复杂度:O(n)
空间复杂度:O(n)

class Solution {
public:
    int numberOfSubarrays(vector& nums, int k) {
        vector csum(nums.size() + 1);
        csum[0] = 1;
        int res = 0, sum = 0;
        for (int i = 0; i = k) res += csum[sum - k];
            ++csum[sum];
        }
        return res;
    }
};

【LeetCode】1248. 统计「优美子数组」

标签:c++   复杂   一个   示例   连续   +=   数列   ber   arrays   

原文地址:https://www.cnblogs.com/galaxy-hao/p/12748667.html


评论


亲,登录后才可以留言!