剑指 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

解析：

　　连续子数组就是组合数 C（n，2），每次记录当前子数组的起点l，和上次子数组的终点r

如果r >= l，则说明这次的子数组和上次的有重复，剪一下就行

否则就是C（n，1） + C（n，2）

class Solution {
public:
    int numSubarrayProductLessThanK(vector<int>& nums, int k) {
        int sum = 1, l = 0, r = -1;
        int len = nums.size();
        int i, ret = 0;
        for(i = 0; i < len; i++)
        {
            sum *= nums[i];
            if(sum >= k)
            {
                if(r >= l) 
                    ret += (i - l) * (i - l - 1) / 2 - (r - l + 1) * (r - l) / 2 + i - r - 1;
                else
                    ret += (i - l) * (i - l - 1) / 2 + i - l;
                r = i - 1;
            }
            while(sum >= k && l <= i)
                sum /= nums[l], l++;
        }
        if(sum < k)
        {
            if(r >= l) 
                ret += (i - l) * (i - l - 1) / 2 - (r - l + 1) * (r - l) / 2 + i - r - 1;
            else
                ret += (i - l) * (i - l - 1) / 2 + i - l;
        }
        return ret;


    }
};