LeetCode No713. 乘积小于 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

提示:

1 <= nums.length <= 3 * 10^4
1 <= nums[i] <= 1000
0 <= k <= 10^6

思路

双指针题目，记录当前双指针子数组的乘积product，如果product<k，那么这个时候满足条件的连续子数组有如下：

nums[high]
nums[high]、nums[high-1]
nums[high]、nums[high-1]、nums[high-2]
...
nums[high]、nums[high-1]、nums[high-2]、...、nums[low]
也就是high-low+1个

即：cnt += (high-low+1)

AC代码

点击查看代码

class Solution {
    public int numSubarrayProductLessThanK(int[] nums, int k) {
        int n = nums.length;
        if( n==0 ) {
            return n;
        }
        int last = 0;
        int product = 1;
        int cnt = 0;
        for(int index=0; index<n; index++) {
            product *= nums[index];
            while( product>=k && last<=index) {
                product /= nums[last];
                last ++;
            }
            cnt += (index-last+1);
        }
        return cnt;
    }
}