题目链接:LCR 009. 乘积小于 K 的子数组 - 力扣(LeetCode)
题目:
输入一个由正整数组成的数组和一个正整数 k,请问数组中有多少个数字乘积小于 k 的连续子数组?例如,输入数组 [10, 5, 2, 6],k 的值为 100,有 8 个子数组的所有数字的乘积小于 100,它们分别是 [10]、[5]、[2]、[6]、[10, 5]、[5, 2]、[2, 6]、[5, 2, 6]。
分析:
代码实现:
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
int n = nums.size();
int product = 1;
int count = 0;
int left = 0;
for (int right = 0; right < n; ++right)
{
product *= nums[right];
while (left <= right && product >= k)
{
product /= nums[left];
++left;
}
count += right - left + 1;
}
return count;
}
};
注意:当 k == 1,且输入数组为只有一个数时,会越界访问 nums[left],所以在 while 循环中需要添加条件: left <= right。