要求
给定一个正整数数组 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
核心代码
class Solution:
def numSubarrayProductLessThanK(self, nums: List[int], k: int) -> int:
if k <= 1:
return 0
left,right = 0,0
product = 1
ret = 0
while right < len(nums):
product *= nums[right]
while product >= k:
product /= nums[left]
left += 1
ret += (right - left + 1)
right += 1
return ret
解题思路：定义左右双指针，右指针先往右走，计算元素的乘积，如果乘积大于等于k则左指针往右走。其中右指针每往右走一步，只要乘积小于k，则计算一次当前的子数组数 right-left+1
