题目:
给你一个整数数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subarray-product-less-than-k
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
这里采用一个双指针滑动窗口的想法。
首先,我们要定义一个双指针,用来确定滑动窗口的左边界和右边界。
其次,我们发现,如果数组中一个长度最大并且符合要求的数组,他的子数组也都符合要求。
就比如[10,5,2,6],他的最大数组为[10,5,2],那么不论是[10]还是[10,5]等等,都是符合要求的。
最后,根据上述的方法,我们只需要不断的给右边界找一个最小的左边界,相当于上一次的窗口,每一次增加的个数为j-i+1,对其进行累加,那么就能得到他的所有子集数,以此方法遍历整个数组即可。
代码:
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
int ans = 0, value = 1;
for(int i = 0, j = 0; i < nums.size(); i++){
value *= nums[i];
while(j <= i && value >= k) value /= nums[j++];
ans += i - j + 1;
}
return ans;
}
};