广州在线网站制作市场seo是什么意思
1 题目:统计「优美子数组」
官方标定难度:中
给你一个整数数组 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] <= 1 0 5 10^5 105
1 <= k <= nums.length
2 solution
代码
class Solution {
public:
int numberOfSubarrays(vector<int> &nums, int k) {unordered_map<int, int> map;map[0] = 1;int s = 0;int sol = 0;for (int i = 0; i < nums.size(); i++) {s += nums[i] % 2;sol += map[s - k];map[s]++;}return sol;
}
};
结果
3 优化
理论上 hash 的时间复杂度为 O(1),但是实际中总是差点意思,所以换成vector 有明显提升。
class Solution {
public:int numberOfSubarrays(vector<int> &nums, int k) {vector<int> map(nums.size() + 1, 0);map[0] = 1;int s = 0;int sol = 0;for (int i = 0; i < nums.size(); i++) {s += nums[i] % 2;if(s >= k)sol += map[s - k];map[s]++;}return sol;
}};