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

分析

以第n个元素为结尾的连续子数组只有n个，利用双指针，右指针指向子数组的结尾，当右指针为right时，如果左指针为left的情况下使得乘积刚好小于k，则以right为结尾的满足条件的子数组共有right-left+1个。

代码

class Solution {
public:
    int numSubarrayProductLessThanK(vector<int>& nums, int k) {
        int sub = 1,count = 0, left = 0;
        for(int i=0;i<nums.size();i++){
            if(nums[i] >= k) {
                left++;
                continue;
            }
            sub = sub * nums[i];
            while(sub >= k && left < i){
                sub = sub/nums[left];
                left++;
            }
            count = count + i - left + 1;
        }
        return count;
    }
};