LeetCode 713. 乘积小于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

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;
    }
};

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