题目描述:
给定一个正整数数组 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的子数组。
解题步骤:
该题可以采用滑动窗口的方法,如果窗口内乘积大于等于k,则将左指针右移,反之将右指针左移。
该题关键在于,如何去确定每次增加的子数组个数。
窗口:[10] 此时增加了 [10]
窗口:[10, 5] 此时增加了 [10,5] — [5]
窗口:[5, 2] 此时增加了 [5, 2] — [2]
窗口:[5,2,6] 此时增加了 [5, 2, 6] — [2, 6] — [6]
根据上面可以看出规律,每次增加的子数组个数,就是当前窗口内的元素个数。
代码:
public int numSubarrayProductLessThanK(int[] nums, int k) {
if(nums.length==0 || k<=1)
return 0;
int val = 1;
int count = 0;
int left = 0;
for (int i = 0; i < nums.length; i++) {
val = val*nums[i];
while(val >= k) {
val = val/nums[left++];
}
count = count + i-left+1;
}
return count;
}