题目
给定一个正整数数组 nums和整数 k ,请找出该数组内乘积小于 k 的连续的子数组的个数。
- 输入与输出
输入:
nums = [10,5,2,6], k = 100
输出: 8
解释: 8 个乘积小于 100 的子数组分别为: [10], [5], [2], [6], [10,5], [5,2], [2,6], [5,2,6]
。 需要注意的是[10,5,2]
并不是乘积小于100的子数组。
思路
滑动窗口的思想,不过与 【剑指Offere II】008 和大于等于 target 的最短子数组 这个题有些不同,需要一点反向的思维。具体操作看代码有详细解释。
代码
class Solution {
public int numSubarrayProductLessThanK(int[] nums, int k) {
// 特殊值判断
if (k <= 1) return 0;
// 定义所需变量
int res = 0;
int left = 0;
int right = 0;
int fac = 1;
// 遍历
while (right < nums.length) {
fac *= nums[right];
// 滑动
while (left <= right && fac >= k) {
fac /= nums[left];
left++;
}
// 窗口大小
// 单纯一个 10 ==> res += 1
// 10 5 1 ==> res += 3
res += right - left + 1;
right++;
}
// 返回结果
return res;
}
}