Skip to content

Latest commit

 

History

History
54 lines (45 loc) · 1.08 KB

File metadata and controls

54 lines (45 loc) · 1.08 KB

乘积小于 K 的子数组

难度:中等

https://leetcode-cn.com/problems/subarray-product-less-than-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

解题

/**
 * 滑动窗口
 * @desc 时间复杂度 O(N) 空间复杂度 O(1)
 * @param nums
 * @param k
 * @returns
 */
export function numSubarrayProductLessThanK(nums: number[], k: number): number {
  const len = nums.length
  let ans = 0
  let prod = 1
  let left = 0
  for (let right = 0; right < len; right++) {
    prod *= nums[right]
    while (left <= right && prod >= k) {
      prod /= nums[left]
      left++
    }
    ans += right - left + 1
  }
  return ans
}