力扣链接:713. 乘积小于 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 * 10^41 <= nums[i] <= 10000 <= k <= 10^6
func numSubarrayProductLessThanK(nums []int, k int) int {
}
func numSubarrayProductLessThanK(nums []int, k int) int {
if k <= 1 {
return 0
}
ans := 0
prod := 1 // 乘积
left := 0
for right, num := range nums {
prod *= num
for prod >= k {
prod = prod / nums[left]
left++
}
ans += right - left + 1
}
return ans
}