LeetCode 713. 乘积小于K的子数组（滑动窗口）

1. 题目

给定一个正整数数组 nums。

找出该数组内乘积小于 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的子数组。
说明:
0 < nums.length <= 50000
0 < nums[i] < 1000
0 <= k < 10^6

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarray-product-less-than-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

• 滑动窗口，连续乘积 < k，右端点向右扩
• 乘积 >= k，左端点向右收缩

class Solution {	//C++
public:
    int numSubarrayProductLessThanK(vector<int>& nums, int k) {
        if(k == 0)
            return 0;
    	int i = 0, j = 0, count = 0, product = 1;
    	while(j < nums.size())
    	{
    		if(product*nums[j] < k)
    		{   //右端点一直乘
    			product *= nums[j];
                count += j-i+1;//以j结尾的子数组个数
                j++;
    		}
    		else// (product*nums[j] >= k)
    		{   //乘积大了，左端点移走
    			product /= nums[i++];//有>=k的数的时候，product可能会变成0
                if(i > j)
                    j++, product=1;//跳过该数字
    		}
    	}
    	return count;
    }
};

304 ms 85.3 MB

class Solution:# py3
    def numSubarrayProductLessThanK(self, nums: List[int], k: int) -> int:
        n = len(nums)
        i, j, count, product = 0, 0, 0, 1
        while j < n:
            if product*nums[j] < k:
                product *= nums[j]
                count += j-i+1
                j += 1
            else:
                product /= nums[i]
                i += 1
                if i > j:
                    j += 1
                    product = 1
        return count

1512 ms 18 MB