给你一个整数数组 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个的乘积。内循环判断如果乘机>=k并且j<=i(注意j<=i的条件,针对每一个值大于k的情况),则让乘积除以第j个值继续判断。内循环结束计数加i-j+1。
class Solution {
public int numSubarrayProductLessThanK(int[] nums, int k) {
int j = 0;
int mul = 1;
int ret = 0;
for(int i = 0; i < nums.length; i++){
mul = mul * nums[i];
while(mul >= k && j <= i){
mul = mul / nums[j];
j++;
}
ret = ret + i - j + 1;
}
return ret;
}
}