713. 乘积小于 K 的子数组

Link

力扣

Problem

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

示例2：

输入：nums = [1,2,3], k = 0
输出：0

提示：

• 1 <= nums.length <= 3 * 104

• 1 <= nums[i] <= 1000

• 0 <= k <= 106

Solution

Go

func numSubarrayProductLessThanK(nums []int, k int) int {
    ans := 0
    for i, j, product := 0, 0, 1; j < len(nums); j++ {
        product *= nums[j]
        for i <= j && product >= k {
            product /= nums[i]
            i++
        }
        ans += j - i + 1
    }
    return ans
}

C++