题目
核心思路
对于求K相关的子数组,一种比较常用的方法是计算前缀和(积),然后通过双指针来做,不过这道题给的数的范围最大的积为1000^50000,实在是太大了,难以实行,官方题解给出了一种转化为log函数相加来防溢出,不过不好理解,若想了解还请看官方题解 713-官解点这里 。
下面来讲解另一种类似的解法,同样是双指针,因为题目给出了数组元素为正整数,故当前乘积 product 乘以/除以一个数是单调的,可以使用滑动窗口来解题。一个指针left标记窗口左端,right标记窗口的右端,乘积大于等于K时,用乘积除以nums[left],然后使得left向前移;若乘积小于K,计算 left 到 right 满足条件的子数组的个数即可。大体思路清楚了,还剩下一点就是求 left 到 right 的子数组的个数待解决,我们通过下图来理解。
完整代码
public int numSubarrayProductLessThanK(int[] nums, int k) {
if (k <= 1)
return 0;
int count = 0;
int left = 0, right = 0, n = nums.length;
int product = 1;//存储当前滑动窗口的乘积
for (; right < n; right++) {
product *= nums[right];
while (product >= k)
product /= nums[left++];
count += right - left + 1;
}
return count;
}
代码比较简洁,不过其中的求每个滑动窗口的子数组个数公式 right - left + 1 不是很好发现,需要仔细研究。