713. 乘积小于 K 的子数组(2022-5-5)
给你一个整数数组 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 * 10^4
- 1 <= nums[i] <= 1000
- 0 <= k <= 10^6
解题思路
简单滑动窗口,不想把问题想的太复杂。
维护一个「滑动窗口」,保证每次循环里窗口都满足乘积小于k的的条件,然后去计算当前窗口的「子数组的个数」
再去进行下一次右端点右移的操作。
对于特殊情况:数组当前元素 大于 k,这时左端点一定会超过右端点,才能保证窗口内的乘积小于k,此时正常计算就好了,当作左端点的重置。
var numSubarrayProductLessThanK = function(nums, k) {
// 滑动窗口
let l = r = 0,count = 0,sum=1
while(r < nums.length){
sum *= nums[r]
while( sum >= k && l <= r) sum /= nums[l++]
count += r - l + 1
r++
}
return count
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subarray-product-less-than-k
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。