给你一个整数数组 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
public static int numSubarrayProductLessThanK(int[] nums, int k) {
int n = nums.length;
int sum = 1;
int left = 0;
// 初始为最大值,方便后序更新
int res = 0;
//特殊情况处理
if (k <= 1) return 0;
for (int right = 0; right < n; right++) {
// 首先计算数组中的和
sum *= nums[right];
// 当满足条件时,记录子数组的长度,注意这里得用while循环一直判断
while (sum >= k) {
// 滑动窗口,不断变更i(子序列的起始位置)
sum /= nums[left++];
}
res += right - left + 1;
}
// 最后收集结果,如果没有没被赋值,则满足条件的子数组的长度为0
return res;
}