天天看點

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      
  • 1 <= nums.length <= 50000
  • 1 <= nums[i] <= 10^5
  • 1 <= k <= nums.length
class Solution {
public:
    int numberOfSubarrays(vector<int>& nums, int k) {
        
        int res=0;
        for(int i=0;i<nums.size();i++)
        {
            int cnt=0;
            for(int j=i;j<nums.size();j++)
            {
                if(nums[j]%2)
                {
                    cnt++;
                }
                if(cnt>k)
                {
                    break;
                }
                if(cnt==k)
                {
                    res++;
                }
            }
        }
        cout<<res<<endl;
        return res;
    }
};