给定一个正整数数组 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 * 104
1 <= nums[i] <= 1000
0 <= k <= 106
这个题一眼就能看出用滑动窗口来做,我试过用暴力解法,会超时。
代码:
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
int n = nums.size();
int front=0, res=0,mul=1;
for(int rear=0;rear<n;rear++)
{
mul*=nums[rear];
while(front<=rear && mul>=k)
{
mul/=nums[front++];
}
res+=rear>=front?(rear-front+1):0;
}
return res;
}
};