【日常训练】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

提示:
1 <= nums.length <= 3 * 10⁴
1 <= nums[i] <= 1000
0 <= k <= 10⁶

代码

package dayLeetCode;

public class dayleetcode713 {
    /**
     *      由于找的是连续子数组的个数，所以可以使用双指针法(滑动窗口)
     *      用变量l，r分别代表子数组的左端和右端，然后将该区间内的数字相乘，查看乘积是否大于k
     *      如果大于k的话就将l左移
     *
     *      如何判断[l,r]中连续子集的个数，以[10,5,2,6],k=100为例
     *      (1)l=0,r=0 tmp=10 1个
     *      (2)l=0,r=1 tmp=50 新增了2个 [5] [10,5]
     *      (3)l=0,r=2 tmp=100 -> l=1,r=2 tmp=10 新增2个 [2][5,2]
     *      (4)l=1,r=3 tmp=60 新增3个 [6][2,6][5,2,6]
     *      通过上述过程我们不难发现每次新增的个数就是 r-l+1
     * @param nums
     * @param k
     * @return
     */

    public int numSubarrayProductLessThanK(int[] nums, int k) {
        // 注意nums中最小为1 当k为0和1时都没有任何子集小于它
        if (k == 0 || k == 1){
            return 0;
        }
        int ans = 0;
        int tmp = 1;
        int l = 0;
        int r = 0;
        for (int i = 0; i < nums.length; i++){
            tmp = tmp * nums[r];
            // l左移
            while (tmp >= k){
                tmp /= nums[l++];
            }
            // 加入[l,r]中子集的个数
            ans += r - l + 1;
            r++;
        }
        return ans;
    }

    public static void main(String[] args) {
        dayleetcode713 obj = new dayleetcode713();
        int[] nums = {10, 5, 2, 6};
        System.out.println(obj.numSubarrayProductLessThanK(nums, 100));
    }
}