【leetcode算法】——713. 乘积小于k的子数组

当代优秀青年2024-10-12 5:05

leetcode算法

滑动窗口
- [713. 乘积小于k的子数组](#713. 乘积小于k的子数组)

滑动窗口

713. 乘积小于k的子数组

滑动窗口：

关键点：

1、不定长滑动窗口

2、cnt += r - l + 1；

解题思路：

nums = [10,5,2,6], k = 100

r = 0， prot = 10，小于k， cnt = 1 [10]
r = 1， prot = 10 * 5，小于k，cnt = r - l +1 = 2 [10,5] [5]
r = 2， prot = 10 * 5 *2，大于k， l++，cnt = r - l + 1 = 2；[5,2] [2]
r = 3, prot = 5 * 2 * 6，小于k， cnt = r - l + 1 = 3 ;[5,2,6] [2,6] [6]
cnt = 8;

c++ 复制代码

class Solution {
public:
    int numSubarrayProductLessThanK(vector<int>& nums, int k) {
        // 初始化数目为0
        int cnt = 0;
        // 乘积为1
        int prot = 1;
        // 左端点
        int l = 0;

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

        for(int r = 0; r<nums.size();++r)
        {
            prot *=nums[r];
            // 当乘积的值大于k，缩小左端点
            while(prot >= k)
            {
                prot /=nums[l++];
            }
            // 求和
            cnt += r - l +1;
        }
        return cnt;
    }
};