题目
给你一个整数数组 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
算法思路
刚开始拿到这个题真不知道怎么下手,考虑了几种方法都不太行,参考了一下评论区,滑动窗口使用数学方法来实现。
滑动窗口,顾名思义就是指定的一段距离通过滑动来解决问题的一种方法,相比较其他方法效率更高,但是算法比较复杂,特别需要注意边界条件,需要的变量也较多。
思路:首先需要定义一个用于存放奇数数字所在索引的数组,滑动窗口的长度为k,是每次的最短长度,然后求得窗口左边距离上一个奇数之间偶数的个数乘以窗口距离下一个奇数之间的偶数个数,这就是本次窗口的「优美子数组」的数目,窗口每次整体右滑动一个位置,直到最后,将每次的数目相加就得到最终结果。
代码
public int numberOfSubarrays(int[] nums, int k) {
int len = nums.length, res = 0, feed = 0, arr[] = new int[len + 2];
for(int i = 0; i < len; i ++) {
// if it is odd
if((nums[i] & 1) == 1) {
arr[++feed] = i; //得到奇数的索引 i
}
}
arr[0] = -1; //左边界
arr[feed + 1] = len; //右边界
for(int i = 1; i + k < feed + 2; i ++) { //滑动窗口
//arr[i] - arr[i - 1] 为左边到上一个奇数的偶数个数
//arr[i + k] - arr[i + k - 1]为右边到下一个为奇数的偶数个数
res += (arr[i] - arr[i - 1]) * (arr[i + k] - arr[i + k - 1]);
}
return res;
}
复杂度
时间复杂度:O(n),空间复杂度:O(n)
注:代码中判断是否为奇数的方法经常会用到,也可以将其定义为一个方法进行调用:
private boolean isOdd(int t) {
return (t & 1) == 1;
}
参考: