Leetcode 713. 乘积小于K的子数组
题目
给定一个正整数数组 nums。
找出该数组内乘积小于 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的子数组。
说明:
- 0 < nums.length <= 50000
- 0 < nums[i] < 1000
- 0 <= k < 10^6
题解
滑动窗口
因为数组中元素均为正整数,因此越乘越大。因此,我们对于每一个下标i,我们找到一个最大的窗口,右边界下标为j,使得乘积小于K,那么以i开头的连续子数组乘积小于K就有j-i+1个。详细过程见代码
代码
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
int i=0,j=0;
int conclude = 1;
int ans=0;
while(i < nums.size()){
if(i>j) j=i;
while(j<nums.size() && conclude*nums[j] < k){
conclude *= nums[j];
j++;
}
ans += j-i;
if(j>i) conclude /= nums[i];
i++;
}
return ans;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subarray-product-less-than-k
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。