给定一个正整数数组 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/problems/ZVAVXX
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路及方法
跟上一道题剑指 Offer II 008. 和大于等于 target 的最短子数组的解法一样,滑动窗口,但是不一样的地方这道题是记录出现次数。
对于乘积小于 k 的连续的子数组的个数,首先我们要排除k=0和k=1的情况,然后每一次移动end指针时,加上满足条件的次数。因为此时从start指针到end指针所有数相乘的积小于k,而且每个数都是正数,对于以下连续子数组:
nums[end]
nums[end-1], nums[end]
nums[end-2], nums[end-1], nums[end]
nums[start], ......, nums[end-2], nums[end-1], nums[end]
乘积均小于k,所以要加上end-start+1。
class Solution {
public int numSubarrayProductLessThanK(int[] nums, int k) {
if (k <= 1) return 0;
int n = nums.length;
// 滑动窗口
int start = 0, end = 0;
// 记录乘积
int product = 1;
int res = 0;
while (end < n) {
product *= nums[end];
// 维护窗口
while (product >= k) {
product /= nums[start];
start++;
}
res += end - start + 1;
end++;
}
return res;
}
}
结果如下:
![](https://img.haomeiwen.com/i6573960/0bb244817f2dc5f7.png)
网友评论