当前位置 博文首页 > debu阿良的博客:LeetCode-Day95(C++) 713. 乘积小于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
双指针
分析
对于每个right,我们需要找到最小的left,满足 ∏ i=leftright nums[i]<k。由于当 left 增加时,这个乘积是单调不增的,因此我们可以使用双指针的方法,单调地移动left。
算法
我们使用一重循环枚举right,同时设置 left 的初始值为 0。在循环的每一步中,表示 right 向右移动了一位,将乘积乘以 ]nums[right]。此时我们需要向右移动left,直到满足乘积小于 k 的条件。在每次移动时,需要将乘积除以 nums[left]。当 left 移动完成后,对于当前的 right,就包含了 \mathrm{right} - right?left+1 个乘积小于 k 的连续子数组。
作者:LeetCode
链接:https://leetcode-cn.com/problems/subarray-product-less-than-k/solution/cheng-ji-xiao-yu-kde-zi-shu-zu-by-leetcode/
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
if (k <= 1) return 0;
int prod = 1, ans = 0, left = 0;
for (int right = 0; right < nums.size(); right++) {
prod *= nums[right];
while (prod >= k) prod /= nums[left++];
ans += right - left + 1;
}
return ans;
}
};
cs