给定一个正整数数组 nums。
找出该数组内乘积小于 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的子数组。
说明:
0 < nums.length <= 50000
0 < nums[i] < 1000
0 <= k < 10^6
思路:
为了避免重复计算,我们采用移动窗口法,这个窗口内数组之积小于k,然后我们一次统计出窗口内的子数组个数,继续一个窗口。
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k)
{
int n = nums.size();
int l = 0, r = 0;
int ans = 0;
int mulRes = 1;
while(r<n)
{
mulRes*=nums[r++];
while(l<r&&mulRes>=k)
mulRes/=nums[l++];
ans+=r-l;
}
return ans;
}
};