713. 乘积小于 K 的子数组

乘积小于 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 的子数组。
示例 2:

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

提示:

1 <= nums.length <= 3 * 104
1 <= nums[i] <= 1000
0 <= k <= 106

思路:

时间复杂度:O(n),空间复杂度O(1)

用滑动窗口法求数量

代码:

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

代码效率:

执行用时:64 ms, 在所有 Go 提交中击败了92.84%的用户
内存消耗:7.2 MB, 在所有 Go 提交中击败了83.70%的用户