乘积小于K的子数组 (滑动窗口leetcode713)-------------------c++实现
题目表述
给定一个正整数数组 nums和整数 k 。
请找出该数组内乘积小于 k 的连续的子数组的个数。
样例
输入: 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的子数组。
条件
1 <= nums.length <= 3 * 104
1 <= nums[i] <= 1000
0 <= k <= 106
思路
用一个总数记录前n个数的乘积,小于k则end右移,结果+=长度,大于则begin右移,end到最后一个点即为结束。
注意点
注意临界条件判断。
ac代码
c++
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
int begin=0,end = 0;
int sum= 1,result=0;
while(end<nums.size()){
sum=sum*nums[end++];
while(sum>=k&&begin<end)
sum=sum/nums[begin++];
result+=(end-begin);
}
return result;
}
};