给你一个整数数组 nums 和一个整数 k ,请你返回子数组内所有元素的乘积严格小于 k 的连续子数组的数目。
示例 1:
输入: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 的子数组。
示例 2:
输入:nums = [1,2,3], k = 0
输出:0
提示:
1 <= nums.length <= 3 * 104
1 <= nums[i] <= 1000
0 <= k <= 106
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
if(k <= 1)
return 0;
int res = 0;
int l = 0;
int r = 0;
long long count = 1;
// 大循环滑动right
while (r < nums.size()) {
count *= nums[r];
// 小循环滑动left 扩大窗口
while(count >= k){
count /= nums[l];
l++;
}
// 每个right有多个子数组 缩小窗口
res += r - l + 1;
r++;
}
return res;
}
};