【剑指 Offer II】9 乘积小于 K 的子数组

题目

• 链接【剑指 Offer II】009 乘积小于 K 的子数组

• 描述

给定一个正整数数组 nums和整数 k ，请找出该数组内乘积小于 k 的连续的子数组的个数。

• 输入与输出

输入: 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的子数组。

思路

滑动窗口的思想，不过与 【剑指Offere II】008 和大于等于 target 的最短子数组 这个题有些不同，需要一点反向的思维。具体操作看代码有详细解释。

代码

class Solution {
    public int numSubarrayProductLessThanK(int[] nums, int k) {
        // 特殊值判断
        if (k <= 1) return 0;
        
        // 定义所需变量
        int res = 0;
        int left = 0;
        int right = 0;
        int fac = 1;

        // 遍历
        while (right < nums.length) {
            fac *= nums[right];

            // 滑动
            while (left <= right && fac >= k) {
                fac /= nums[left];
                left++;
            }

            // 窗口大小
            // 单纯一个 10 ==> res += 1
            // 10 5 1 ==> res += 3
            res += right - left + 1;

            right++;
        }

        // 返回结果
        return res;
    }
}