[LeetCode]剑指 Offer II 009. 乘积小于 K 的子数组

小于 K 的子数组

Posted by JinFei on November 12, 2021

题目描述

给定一个正整数数组 nums和整数 k ,请找出该数组内乘积小于 k 的连续的子数组的个数。

Example1:

输入: 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的子数组。

Example2:

输入: nums = [1,2,3], k = 0 输出: 0

Note:

  • 1 <= nums.length <= 3 * 10^4
  • 1 <= nums[i] <= 1000
  • 0 <= k <= 10^6

解题思路

class Solution {
public:
    int numSubarrayProductLessThanK(vector<int>& nums, int k) {
        int right = 0;
        int left = 0;
        int res = 0;
        long pro = 1;
        for(; right < nums.size(); right++){
            pro *= nums[right];
            while(left <= right && pro >= k){
                pro /= nums[left];
                left++;
            }
            res += right >= left ? right - left + 1 : 0;
        }
        return res;
    }
};