给你一个整数数组 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
思路:
【n1,n2,n3,n4,n5】 为例
条件1:N1--N(m)子数组的总个数 = 以N1结尾的子数组个数 + N2结尾的子数组个数+ .... + N(m)结尾的子数组个数。
条件2:N(m)结尾的子数组个数 = N(L) 到 N(m)的最远距离 (即如果再乘以N(L-1),乘积肯定大于等于K)
根据条件1和条件2:
我们只要定义left和right指针,求得以每个位置N(right)结尾,到最远距离N(left)的子数组个数,就可以得出整个数组的满足条件的子数组个数。
起始位置left和right都为0。
如果乘积大于等于K,那么left++;
当left超过right时候,说明当前位置的数字本身就大于K,导致left移动位置超过right,那么就将right和left进行对齐。
如果乘积小于K,可以得出结论:当前right的最长度是到达left,以right结尾的子数组个数是right - left + 1.
然后right++;
class Solution {
public int numSubarrayProductLessThanK(int[] nums, int k) {
int left = 0;
int right = 0;
int mul = nums[left];
int result = 0;
while (right < nums.length) {
if (mul < k) {
//从left----right的数组乘积小于K
result += 1 + right - left;
//那么right继续往右移动
right++;
if(right >= nums.length)
break;
mul = mul * nums[right];
} else {
mul /= nums[left];
left++;
//当左指针移动超过right的时候,将右指针和左指针对齐,出现这个情况的场景是某个位置的数字大于K。
if (left > right) {
right = left;
if(right >= nums.length)
break;
mul = nums[right];
}
}
}
return result;
}
}