给定一个正整数数组 nums。
找出该数组内乘积小于 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的子数组。
说明:
0 < nums.length <= 50000
0 < nums[i] < 1000
0 <= k < 10^6
通过次数5,585提交次数15,949
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subarray-product-less-than-k
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路一 :暴力超时 可以过 75/84 个点
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
int len = nums.size();
if(len == 0)
return 0;
int cnt = 0;
vector<int> dp(len+1,1);
int tmp = 1;
for(int i=0;i<len;i++){
tmp = 1;
for(int j=i;j<len;j++){
tmp = tmp * nums[j];
if(tmp<k)
cnt++;
else
break;
}
}
return cnt;
}
};
思路二:双指针
如果当前乘积小于k,那么right左边含right的所有子串乘积一定都小于k;如果新来个right后乘积大于等于k了,那么就缩left,直到乘积再次小于right,此时left与right之间所有含right的子串的乘积一定都小于k。
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
int len = nums.size();
if(len == 0)
return 0;
if(k<=1)
return 0;
int cnt = 0;
int tmp = 1;
int left =0;
int right = 0;
while(right<len){
tmp = tmp * nums[right];
while(tmp>=k){
tmp /= nums[left++];
}
cnt += right - left + 1;
right++;
}
/*for(int i=0;i<len;i++){
tmp = 1;
for(int j=i;j<len;j++){
tmp = tmp * nums[j];
if(tmp<k)
cnt++;
else
break;
}
} */
return cnt;
}
};