乘积小于 K 的子数组

92 阅读1分钟

给你一个整数数组 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 的子数组。

双指针 滑动窗口

计算窗口中间的乘积

class Solution { public int numSubarrayProductLessThanK(int[] nums, int k) {

    int left = 0;
    int right = 0;
    int tmp = 1;
    int ans = 0;

    int numsLen = nums.length;

    // if (k == 0 || k==1){
    //     return 0;
    // }
            if (k <=1){
        return 0;
    }
    

    while(right<numsLen) {
        tmp *= nums[right];
        while (tmp >= k) {
            tmp /= nums[left];
            left++;
        }
        //中间有这么多种组合
        ans += right - left + 1;
        right++;
    }
    return ans;
}

}