Count and print the number of (contiguous) subarrays where the product of all the elements in the subarray is less thank.
Example 1:
Input:
nums = [10, 5, 2, 6], k = 100
Output:
8
Explanation:
The 8 subarrays that have product less than 100 are: [10], [5], [2], [6], [10, 5], [5, 2], [2, 6], [5, 2, 6].
Note that [10, 5, 2] is not included as the product of 100 is not strictly less than k.
for window (5, 2), when 6 is introduced, it add 3 new subarray: (5, (2, (6)))
(6)
(2, 6)
(5, 2, 6)
class Solution:
def numSubarrayProductLessThanK(self, nums: List[int], k: int) -> int:
s,e,l,p,res = 0,0,len(nums),1,0
while e < l:
p *= nums[e]
e += 1
while p >= k and s < e:
p /= nums[s]
s += 1
res += e-s
return res