力扣713. 乘积小于 K 的子数组
题目链接:713. 乘积小于 K 的子数组
给你一个整数数组 nums 和一个整数 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
解题思路:
暴力从左到右模拟数组,移动右端点,一直乘,直到>=k,此时就需要删除左端点元素,当然我们的循环还要 && L <= R 防止越界,当循环出来后就代表乘积一定合法
因为上一位置合法,这一位置又合法所以,ai,ai-1 ai,就代表两个集合
所以我们每次只需要加上包含最右元素的新集合即可算出答案
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
int l = 0, r = 0, ans = 0;
long long c = 1;
//遍历数组
for(; r < nums.size(); r++)
{
//每次乘上新的元素
c *= nums[r];
//当乘积>=k,并且l<=r(防止最后一个元素导致越界)
while(c >= k && l <= r)
{
//删除左边的选段
c /= nums[l];
l++;
}
//当循环结束之后乘积一定满足题目要求
//因为上一位置合法,这一位置又合法所以,ai,ai-1 ai,就代表两个集合
//每次只需要加上包含最右元素的新集合即可算出答案
ans += r - l + 1;
}
return ans;
}
};