713. 乘积小于 K 的子数组
给你一个整数数组 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
思路:滑动窗口。设置两个指针i和j,先固定左端点i,枚举右端点j,当一个窗口内的子数组乘积大于等于k时,条件开始不满足,这时候可以变动窗口,也就是把左端点的数除去,然后左端点右移,满足条件的窗口,子数组数量为j-i+1,要每次加上这个数量
class Solution {
public int numSubarrayProductLessThanK(int[] nums, int k) {
int cnt = 0;
int prod = 1;
int i = 0;
for(int j = 0; j < nums.length; j++){
prod *= nums[j];
while(i <= j && prod >= k){
prod /= nums[i];
i++;
}
cnt += j - i + 1;
}
return cnt;
}
}