标签:nums int sum ++ 数组 1248 LeetCode size
题目
给你一个整数数组 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
思路一:数学
代码
时间复杂度:O(n)
空间复杂度:O(n)
class Solution {
public:
int numberOfSubarrays(vector<int>& nums, int k) {
int size = nums.size(), cnt = 0, res = 0;
vector<int> odd(size + 2);
odd[0] = -1;
for (int i = 0; i < size; ++i) {
if (nums[i] & 1) odd[++cnt] = i;
}
odd[++cnt] = size;
for (int i = 1; i + k <= cnt; ++i) {
res += (odd[i] - odd[i - 1]) * (odd[i + k] - odd[i + k - 1]);
}
return res;
}
};
思路二:前缀和
代码
时间复杂度:O(n)
空间复杂度:O(n)
class Solution {
public:
int numberOfSubarrays(vector<int>& nums, int k) {
vector<int> csum(nums.size() + 1);
csum[0] = 1;
int res = 0, sum = 0;
for (int i = 0; i < nums.size(); ++i) {
sum += nums[i] & 1;
if (sum >= k) res += csum[sum - k];
++csum[sum];
}
return res;
}
};
标签:nums,int,sum,++,数组,1248,LeetCode,size 来源: https://www.cnblogs.com/galaxy-hao/p/12748667.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。