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;
}
完----------------------------------------------------------