题目:
给定一个正整数数组 nums和整数 k ,请找出该数组内乘积小于 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的子数组。
示例 2:
输入: nums = [1,2,3], k = 0
输出: 0
提示:
1 <= nums.length <= 3 * 104
1 <= nums[i] <= 1000
0 <= k <= 106
--------------------------
思考:
这道题乍一看满足滑窗的条件,让我们找小于K的连续子数组的个数,
但这不是求最大最小滑窗的长度,而是要求子数组的个数。有点不满足公式啊?
别着急否定,让我们来画个图考虑下滑窗右边界移动这个操作与滑窗内子数组个数的关系吧!
通过画图我们发现:
窗口每次移动后,ret都可以增加 right - left + 1个子数组。
这不就可以通过滑窗来解题了么?
但这里要注意一点:
如果数组中某个数字比K还大,则left会超过right,以保证有值,此时窗口长度为-1,无需计算。
另外:由于K<=10 ^ 6,nums[i]<=1000, 10 ^ 9 小于Integer.MAX_VALUE,所以Java使用int类型不会越界。
------------------
class Solution {//双指针 + 滑动窗口
public int numSubarrayProductLessThanK(int[] nums, int k) {
int left = 0;
int res = 0;//结果集
int productSum = 1;//乘积之和
for (int right = 0; right < nums.length; right++) {
productSum *= nums[right];
while (left <= right && productSum >= k) {//滑动窗口移动条件
productSum /= nums[left++];
}
if (left <= right) {//退出while循环后 记录当次滑动窗口内的 子数组个数
res += right - left + 1;
}
}
return res;
}
}