题目描述:
给你一个整数数组 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
这道题emm,我也没想出来,
List<Integer> loc = new ArrayList<>();
loc.add(-1);
for (int i = 0; i < nums.length; i++) {
if (nums[i] % 2 == 1) {
loc.add(i);
}
}
loc.add(nums.length);
int res = 0;
for (int i = k; i < loc.size() - 1; i++) {
int l = loc.get(i-k+1) - loc.get(i-k);
int r = loc.get(i+1) - loc.get(i);
res += l * r;
}
return res;