刷题笔记——乘积小于 K 的子数组

给你一个整数数组 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 的子数组。

def solve(nums,k):
    ans=0
    mul=1
    i=0
    for j,num in enumerate(nums):
        mul*=num
        while i<=j and mul>=k:
            mul//=nums[i]
            i+=1
        ans+=j-i+1
    return ans