Subarray Product Less Than K

最新推荐文章于 2024-07-10 21:00:42 发布

转载最新推荐文章于 2024-07-10 21:00:42 发布 · 155 阅读

本文介绍了一种算法，用于计算给定整数数组中所有连续子数组的数量，这些子数组的元素乘积小于指定阈值k。通过使用滑动窗口的方法，确保了高效地遍历数组并计算符合条件的子数组数量。

Your are given an array of positive integers nums.

Count and print the number of (contiguous) subarrays where the product of all the elements in the subarray is less than k.

Example 1:

Input: nums = [10, 5, 2, 6], k = 100
Output: 8
Explanation: The 8 subarrays that have product less than 100 are: [10], [5], [2], [6], [10, 5], [5, 2], [2, 6], [5, 2, 6].
Note that [10, 5, 2] is not included as the product of 100 is not strictly less than k.

Note:

0 < nums.length <= 50000.0 < nums[i] < 1000.0 <= k < 10^6.

class Solution {
public:
    int numSubarrayProductLessThanK(vector<int>& nums, int k) {
        int cnt = 0;
        int multi = 1;
        queue<int> que;
        for(int i=0; i<nums.size(); i++)
        {
            while(nums[i]*multi>=k&&que.size())
            {
                multi/=que.front();
                que.pop();
            }
            
            if(nums[i]*multi<k)
            {
                multi *= nums[i];
                que.push(nums[i]);
            }
            
            cnt += que.size();
        }
        
        return cnt;
    }
};