每日刷题第7天 2021.1.2

乘积小于k的子数组

• 难度：中等

题目

• 给定一个正整数数组 nums和整数 k ，请找出该数组内乘积小于 k 的连续的子数组的个数。

示例

输入: 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的子数组。

输入: nums = [1,2,3], k = 0
输出: 0

提示

• 1 <= nums.length <= 3 * 104
• 1 <= nums[i] <= 1000
• 0 <= k <= 106

解法

• 加起来是小于的，那么单个就是小于的，直接计算长度即可。

/**
 * @param {number[]} nums
 * @param {number} k
 * @return {number}
 */
var numSubarrayProductLessThanK = function(nums, k) {
  if (k == 0) {
    return 0;
  }
  // 类似于几个数的和
  let start = 0;
  let end = 0;
  let ans = 1;
  let len = 0;
  let flag = false;
  while (end < nums.length) {
    // 常规routine
    ans *= nums[end];
    while (ans >= k && start <= end) {
      ans /= nums[start];
      start++;
    }
    // 小于k分情况
    if (start <= end) {
      len += end - start + 1;
    }
    end++;
  }
  return len;
};

附录