题目
给你一个整数数组 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^4 1 <= nums[i] <= 1000 0 <= k <= 10^6 来源:力扣(LeetCode)
解题思路
??这是一个非常经典的滑动窗口的题,我们可以枚举窗口的长度从1到len(nums),每种窗口遍历一边数组。
class Solution:
def numSubarrayProductLessThanK(self, nums: List[int], k: int) -> int:
frame,count,n=1,0,len(nums)
while True:
flag,s=0,1
for i in range(n-frame+1):
if i==0:
for j in range(i,i+frame):
s*=nums[j]
else:
s//=nums[i-1]
s*=nums[i+frame-1]
if s<k:
count+=1
flag=1
frame+=1
if not flag:
break
return count
??这样做的缺点是显而易见的,每次回到开始重新计算乘积将会有大量的重复计算,所以需要改变思路,滑动的窗口不能是一层不变的,需要在遍历数组的过程中像蠕虫一样自由伸缩,向前移动。
class Solution:
def numSubarrayProductLessThanK(self, nums: List[int], k: int) -> int:
s,count,j=1,0,0
for i in range(len(nums)):
s*=nums[i]
while j<=i and k<=s:
s//=nums[j]
j+=1
count+=i-j+1
return count
|