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

思路其实一开始就想明白了
从数组中去遍历到k个奇数，对于每一组满足的，求解左右两边的偶数长度
例如 nums = [2,2,2,1,2,2,1,2,2,2], k = 2
奇数段就是1221 左右偶数段都是222
因此答案应该是排列组合 奇数段个数+左右偶数段个数+左右偶数段乘积=1+3+3+3*3=16种
基于这个思路写了如下代码

class Solution {
public:
        int numberOfSubarrays(vector<int>& nums, int k) {
		int left=0,right=0,count=0,leftnum=0,rightnum=0;
		int sol=0;
		int n=nums.size();
		if(n==0)
		return 0;
		for(int i=0;i<n;i++)
		{
			count=0;
			if(nums[i]%2==1)
			{
				left=i;count++;right=left+1;leftnum=0;rightnum=0;
				for(right=left+1;right<n&&count<k;right++)
				{
					if(nums[right]%2==1)
					count++;
				}
				if(count==k)
				{
					sol++;
					for(int j=left-1;j>=0;j--)
					{
						if(nums[j]%2==0)
						leftnum++;
						if(nums[j]%2==1)
						break;	
					}	
					for(int j=right;j<n;j++)
					{
						if(nums[j]%2==0)
						rightnum++;	
						if(nums[j]%2==1)
						break;
					}	
					sol+=leftnum;
					sol+=rightnum;
					if(leftnum!=0&&rightnum!=0)
					sol+=leftnum*rightnum;	
				}
			}	
		} 
		return sol;
    }
};

不出所料 在面对26组的大量数据时 妥妥的超时了
5555~

后来看到了一种蛮独特的解法
把连续偶数合并 以奇数分隔开
例如 nums = [2,2,2,1,2,2,1,2,2,2], k = 2
合并为[4,3,4] 也就是排列组合的方式
答案就是4*4=16种

    int numberOfSubarrays(vector<int>& nums, int k) {
	int preEven=0,j=0;
	int n=nums.size();
        int arr[99999]={0};
        for(int i=0;i<n;i++){
            if (nums[i]%2==0){
                preEven++;
            }else{
                arr[j++]=preEven+1;
                preEven=0;
            }
        }
        arr[j++]=preEven+1;
        //for(int i=0;i<j;i++)
       // cout<<arr[i]<<endl;
        int count=0;
        for (int i=0;i<j-k;i++){
            count+=(arr[i]*arr[i+k]);
        }
        return count;
    }

完----------------------------------------------------------