leetcode 713. 乘积小于K的子数组（C++、python）

给定一个正整数数组 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

C++

class Solution {
public:
    int numSubarrayProductLessThanK(vector<int>& nums, int k) 
    {
        int sum=1;
        int n=nums.size();
        int res=0;
        int left=0;
        for(int i=0;i<n;i++)
        {
            sum*=nums[i];
            while(sum>=k && left<=i)
            {
                sum/=nums[left];
                left++;
            }
            res+=i-left+1;
        }
        return res;
    }
};

python

class Solution:
    def numSubarrayProductLessThanK(self, nums: List[int], k: int) -> int:
        res=0
        su=1
        left=0
        n=len(nums)
        for i in range(n):
            su*=nums[i]
            while su>=k and left<=i:
                su//=nums[left]
                left+=1
            res+=i-left+1
        return res