剑指 Offer II 009. 乘积小于 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ZVAVXX
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
分析
跟上题类似,不过是从求最小的滑动窗口变成了求滑动窗口的组合数目。
题解中利用了 right - left + 1 来增加 ans,含义是:如果滑动窗口中数字的积小于 k,就获取所有以 right 为新的右端点的滑动窗口的新增数字组合数,以 [10, 5, 1, 2, 6] 为例,第一个是 10,对应 right 是 0,ans 加了 1 即组合 [10],第二个数字是 5,对应 right 是 1,ans 加了 2,这个 2 表示 [10, 5] 以及 [5] 两个组合,第三个数字是 1,对应 right 是 2,ans 加了 3,对应 [10, 5, 1]、[5, 1]、[1] 这三个组合;以上是积小于 k 的情况,当积大于等于 k,left 右移,以 right = 3 为例,left 移到 1, right - left + 1 = 3 对应新增的组合是 [5, 1, 2]、[1, 2]、[2]。可以理解为前面的组合已经记录好了,新增一个数会多right - left + 1 个组合。
题解(Java)
class Solution {
public int numSubarrayProductLessThanK(int[] nums, int k) {
int left = 0;
int product = 1;
int ans = 0;
for (int right = 0; right < nums.length; right++) {
product *= nums[right];
while (left <= right && product >= k) {
product /= nums[left++];
}
ans += right - left + 1;
}
return ans;
}
}