leetcode-713. 乘积小于K的子数组

713. 乘积小于K的子数组

题目描述

给定一个正整数数组 nums。

找出该数组内乘积小于 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的子数组。

说明:

0 < nums.length <= 50000
0 < nums[i] < 1000
0 <= k < 10^6

思路解释

双指针法，具体看代码注释。

代码

class Solution {
public:
    int numSubarrayProductLessThanK(vector<int>& nums, int k) {
        // 双指针法
        if (k <= 1) 
            return 0;
        int left = 0, right = 0;
        int len = nums.size();
        int ans = 0, mul = 1;
        for (;right < len; right++) {
            // 串中新增一个右指针指向的值
            mul *= nums[right];
            // 这里将串的范围缩小到小于K中，此时该串中所有的子串都小于k
            while(mul >= k) {
                mul /= nums[left];
                left++;
            }
            // 为什么是right - left + 1，比如示例1中，当前串为[5, 2, 6]时，而当前right指向6（相应的left指向5），所以就要加上[6], [2,6], [5,2,6]，因为6是新加进来的。
            ans += right - left + 1;     
        }
        return ans;
    }
};