剑指 Offer II 009. 乘积小于 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 * 104
1 <= nums[i] <= 1000
0 <= k <= 106

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/ZVAVXX
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路及方法

跟上一道题剑指 Offer II 008. 和大于等于 target 的最短子数组的解法一样,滑动窗口,但是不一样的地方这道题是记录出现次数。

对于乘积小于 k 的连续的子数组的个数,首先我们要排除k=0和k=1的情况,然后每一次移动end指针时,加上满足条件的次数。因为此时从start指针到end指针所有数相乘的积小于k,而且每个数都是正数,对于以下连续子数组:
nums[end]
nums[end-1], nums[end]
nums[end-2], nums[end-1], nums[end]
nums[start], ......, nums[end-2], nums[end-1], nums[end]
乘积均小于k,所以要加上end-start+1。

class Solution {
    public int numSubarrayProductLessThanK(int[] nums, int k) {
        if (k <= 1) return 0;
        int n = nums.length;

        // 滑动窗口
        int start = 0, end = 0;
        // 记录乘积
        int product = 1;
        int res = 0;
        while (end < n) {
            product *= nums[end];
            // 维护窗口
            while (product >= k) {
                product /= nums[start];
                start++;
            }
            res += end - start + 1;
            end++;
        }
        
        return res;
    }
}

结果如下:

你可能感兴趣的:(剑指 Offer II 009. 乘积小于 K 的子数组)