给你一个整数数组 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 * 10^4
1 <= nums[i] <= 1000
0 <= k <= 10^6
解法:双指针滑动窗口
class Solution {
public int numSubarrayProductLessThanK(int[] nums, int k) {
if (k <= 1) {
return 0;
}
int count = 0;
int left = 0;
int right = 0;
int multify = 1;
int n = nums.length;
while (right < n) {
multify *= nums[right];
while (multify >= k) {
multify /= nums[left];
left++;
}
// 每次右指针位移到一个新位置,应该加上 x 种数组组合:
// nums[right]
// nums[right-1], nums[right]
// nums[right-2], nums[right-1], nums[right]
// nums[left], ......, nums[right-2], nums[right-1], nums[right]
//共有 right - left + 1 种
count += right - left + 1;
right++;
}
return count;
}
}
复杂度分析
本题采用的是双指针滑动窗口,大循环是右指针的移动,内部小循环是左指针的移动。
时间复杂度:O(n),其中 n 是数组 nums 的长度,右指针遍历一遍数组,左指针紧随其后最多遍历一遍数组;
空间复杂度:O(1),只创建了有限的几个常量变量作为记录。