【每日一题】LeetCode 1248. 统计优美子数组

每日一题，防止痴呆 = =

一、题目大意

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/count-number-of-nice-subarrays

二、题目思路以及AC代码

因为前两天做过一道双指针的题，这次我首先就想到了。对于确定优美子数组，无非就是让你确定左右边界，然后考虑该题。

以这个例子来说，其中k=2，我们可以首先找到所有奇数的位置，也就是图中标明的①，②，③，如果对于k=2来说，我们首先可以找到的优美子序列是[①，②]，然后①的左边有3个选择，加上它本身就是4个，②的右边有3个选择，加上他本身也是4个，所以总共包含这两个奇数的优美子序列就是4×4=16个，然后我们再考虑[②，③]，同样，②的左边有2个选择，加它自身就是3个选择，③右边有2个选择，加自身就是3个选择，所以包含②③着两个奇数的优美子序列共3×3=9个，所以总共包含两个奇数的优美子序列就是9+16=25个。

这是一个数学的解法，当然上面只是引导一下你的思路，具体可以看我后面的代码。

然后我在题解中发现了另一种解法，当然在思考的时候也考虑过，就是用前缀和的方法。同样用上述的例子来说明。

图中一共有三个数组，其中①是原始数组a，②是前缀和数组pre，③是计次数组cnt。
pre[i]表示截止i之前，共有多少个奇数，cnt[i]表示pre[i]出现了几次。

我们可以这样考虑，有了pre数组之后，假设[j, i]是一个优美子数组，那么其一定满足: pre[i] - pre[j-1] = k，那么我们如果想知道以第i个数作为结尾，共有多少个优美子数组，其实只需要求解pre[j-1] = pre[i] - k共出现了多少次即可，但是如果单纯的遍历，会造成O(n^2)的复杂度，这里就可以使用计次数组去把它化简为O(n)，其实就是把查找从O(n)简化为了O(1)，具体可以看我后面的代码，但是那不是我写的 = =，我只写了上面一种。

下面给出两者的AC代码：

一、数学解法

class Solution {
public:
    int numberOfSubarrays(vector<int>& nums, int k) {
        vector<int> pos;
        pos.push_back(-1);
        
        int idx = 0;
        for (auto num: nums) {
            if (num & 1) {
                pos.push_back(idx);
            }
            idx++;
        }
        pos.push_back(idx);

        int p_size = pos.size();
        if (p_size < k + 2) return 0;

        int res = 0;
        for (int i=0;i<p_size-k-1;i++) {
            res += (pos[i+1]-pos[i]) * (pos[i+k+1]-pos[i+k]);
        }

        return res;
    }
};

二、前缀和解法

class Solution {
    vector<int> cnt;
public:
    int numberOfSubarrays(vector<int>& nums, int k) {
        int n = (int)nums.size();
        cnt.resize(n + 1, 0);
        int odd = 0, ans = 0;
        cnt[0] = 1;
        for (int i = 0; i < n; ++i) {
            odd += nums[i] & 1;
            ans += odd >= k ? cnt[odd - k] : 0;
            cnt[odd] += 1;
        }
        return ans;
    }
};

作者：LeetCode-Solution

如果有问题，欢迎大家指正！！！