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

题目

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

提示：
1 <= nums.length <= 50000
1 <= nums[i] <= 10^5
1 <= k <= nums.length

一、动态规划（超时）

1.确定状态：建立dp数组用来表示当前位置之前共有几个奇数
2.状态转移：如果当前位置为奇数，dp[i] = dp[i - 1] + 1
3.边界确定：初值都赋为0

代码：

class Solution {
    int n;
    int dp[50005];
public:
    int numberOfSubarrays(vector<int>& nums, int k) {
        n = nums.size();

        int ans = 0;
        for(int i = 1; i <= n; i++){
            if(nums[i - 1] & 1) dp[i] = dp[i - 1] + 1;
            else dp[i] = dp[i - 1];
            for(int j = 0; j < i; j++)
                if(dp[i] - dp[j] == k) ans++;
        }

        return ans;
    }
};

时间复杂度达到了O(n²)，超时。

二、滑动窗口（通过）

1.初始化left，right指针，通过不断向右移动right指针来扩大窗口，不断更新窗口里所含有的奇数个数。
2.当奇数个数等于k：
计算left指针与窗口中第一个奇数之间偶数的个数leftDis
计算right指针与下一个奇数之间偶数的个数rightDis
此时left指针可以停放的位置有当前位置和leftDis中的任意一个，right指针可以停放的位置有当前位置和rightDis中的任意一个。更新答案ans = (leftDis + 1) * (rightDis + 1)

代码：

class Solution {
    int n;
public:
    int numberOfSubarrays(vector<int>& nums, int k) {
        n = nums.size();

        int left = 0;
        int right = 0;
        int cnt = 0, ans = 0;
        while(right < n){
            if(nums[right++] & 1) cnt++;
            if(cnt == k){
                int tmp = right;
                while(right < n && (nums[right] & 1) == 0) right++;
                int rightDis = right - tmp;

                tmp = left;
                while((nums[left] & 1) == 0) left++;
                int leftDis = left - tmp;

                ans += (leftDis + 1) * (rightDis + 1);
                left++;
                cnt--;
            }
        }       
        return ans;
    }
};

时间复杂度：O(n)