【Leetcode】713. Subarray Product Less Than K

题目地址：

https://leetcode.com/problems/subarray-product-less-than-k/

给定一个正整数长$n$的数组$A$，再给定一个非负整数$k$，问$A$中有多少个子数组的乘积是小于$k$的。

思路是双指针。开左右两个指针$j$和$i$，再用一个变量维护$\prod A[j:i]$，枚举子数组右端点，如果乘积小于$k$，那么以$A[i]$为右端点的满足条件的子数组的个数就是$i-j+1$；否则右移左指针直到乘积小于$k$。注意，每次枚举完右端点之后，左指针不必回退，因为这只会导致乘积更大。代码如下：

class Solution {
 public:
  int numSubarrayProductLessThanK(vector<int>& a, int k) {
    int res = 0;
    for (int i = 0, j = 0, prod = 1; i < a.size(); i++) {
      prod *= a[i];
      while (j <= i && prod >= k) prod /= a[j++];
      res += i - j + 1;
    }
    return res;
  }
};

时间复杂度$O(n)$，空间$O(1)$。