题目
给你一个整数数组 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/subarray-product-less-than-k
解题
双指针法/滑动窗口,第二个指针控制摘除的数字。摘除数字之后可以避免子数组计算重复的问题。
var numSubarrayProductLessThanK = function(nums, k) {
if(k == 0 || k==1 ) return 0;
let l = 0; //摘除元素数量
let prod = 1; //乘积
let res = 0; //子数组个数
for( let i = 0; i<nums.length;i++){
prod *= nums[i];
while(prod >= k){ //每次大于k就开始摘除前面的,队列想法,先入先出
prod /= nums[l++];
}
res += i + 1 -l; //子数组数正好是等于数组中元素的数量,i+1是全部元素的数量,l是摘除元素的数量
}
return res
};