LC713 乘積小于k的子數組
1 題目
給你一個整數數組 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 * 1041 <= nums[i] <= 10000 <= k <= 106
2 解答
class Solution:
def numSubarrayProductLessThanK(self, nums: List[int], k: int) -> int:
n = len(nums)
left = 0
sum = 1
ans = 0
if k <= 1:
return 0
for right,value in enumerate(nums):
sum = sum * value
while sum >= k :
sum = sum // nums[left]
left += 1
ans = ans + right - left + 1
return ans

浙公網安備 33010602011771號