713. 乘积小于 K 的子数组
给你一个整数数组 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
对于本题是求一个符合条件的连续子数组,看到这我们联想到力扣209. 长度最小的子数组,这题与之类似同样是求一个连续的子数组。运用双指针滑动窗口,同样枚举右指针,在出现乘积严格小于 k时做处理。
代码如下:
public int numSubarrayProductLessThanK(int[] nums, int k) {
if(k<=0){// 前提判断
return 0;
}
// 设置左指针,返回结果,乘积
int left = 0,res=0,product=1;
// 枚举右指针
for(int right=0;right<nums.length;right++){
// 计算乘积
product *= nums[right];
// 当乘积大于等于K时左处理,由于是移动左指针所以此处增加了对左指针的限制 <= 右指针
while(product >= k && left<= right){
// 移动左指针,乘积则对应反向操作进行除法
product /= nums[left];
left++;
}
// 乘积小于k时,此时窗口左右指针为left、right,这个时候就需要计算连续子数组个数了
res += (right-left+1);
}
return res;
}
运行结果如下:
对于上述解法中 res += (right-left+1),可能开始看题解不是很明白。由于我们使用的res为结果的累加之和,所以在处理当前元素时其前面的所有符合情况的子数组个数已经求出,只需要再求增加当前元素的子数组个数即可。
假如:[left,…,pre,cur]符合条件并且已经知道了[left,pre]的子数组个数res。当枚举到cur且符合条件需要计算子数组个数,此时窗口为[left,cur],其实相对于[left,…,pre]就是增加了单个cur元素,那么新增的子数组(连续的)就是[left,…,cur]、[left+1,…,cur]、…、[cur],也就是right-left+1个。
个人见解不喜勿喷!!!