此题来源:LeetCode 1248
- 题目简述:
给你一个整数数组 nums 和一个整数 k。如果某个 连续 子数组中恰好有 k 个奇数数字,我们就认为这个子数组是「优美子数组」。请返回这个数组中「优美子数组」的数目。 - 思路:
我们需要知道区间[l,r],在这个区间内整数正好为k个,这样区间究竟有多少个。那很明显我们需要知道某个区间的第一个奇数离它在数组内的前一个奇数中有多少个偶数n,以及最后一个奇数离它在数组内的后一个奇数中间有多少个偶数m,而(n+1)*(m+1)就是这个组优美子数组的总共的区间个数(因为他们本身也算一种情况),然后所有组的加起来。 - 方法:
所以我们只需要记下所有奇数的下标就可以了。 - 边界处理:
因为有可能这个区间的第一个奇数前没有奇数了,或者最后一个奇数后没有奇数了,所以需要对这两个边界值做处理,往奇数下标数组里头尾插入-1和n,具体见代码。
AC代码:
class Solution {
public:
int numberOfSubarrays(vector<int>& nums, int k) {
int n = (int)nums.size();
int odd[n + 2], ans = 0, cnt = 0;
for (int i = 0; i < n; ++i) {
if (nums[i] & 1) odd[++cnt] = i;
}
odd[0] = -1, odd[++cnt] = n;
for (int i = 1; i + k <= cnt; ++i) {
ans += (odd[i] - odd[i - 1]) * (odd[i + k] - odd[i + k - 1]);
}
return ans;
}
};