1. 题目
给定一个正整数数组 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subarray-product-less-than-k
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 解题
- 滑动窗口,连续乘积 < k,右端点向右扩
- 乘积 >= k,左端点向右收缩
class Solution { //C++
public:int numSubarrayProductLessThanK(vector<int>& nums, int k) {if(k == 0)return 0;int i = 0, j = 0, count = 0, product = 1;while(j < nums.size()){if(product*nums[j] < k){ //右端点一直乘product *= nums[j];count += j-i+1;//以j结尾的子数组个数j++;}else// (product*nums[j] >= k){ //乘积大了,左端点移走product /= nums[i++];//有>=k的数的时候,product可能会变成0if(i > j)j++, product=1;//跳过该数字}}return count;}
};
304 ms 85.3 MB
class Solution:# py3def numSubarrayProductLessThanK(self, nums: List[int], k: int) -> int:n = len(nums)i, j, count, product = 0, 0, 0, 1while j < n:if product*nums[j] < k:product *= nums[j]count += j-i+1j += 1else:product /= nums[i]i += 1if i > j:j += 1product = 1return count
1512 ms 18 MB