给你一个整数数组 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 * 10 4 ^4 4
1 <= nums[i] <= 1000
0 <= k <= 10 6 ^6 6
滑动窗口,当窗口内的元素之积小于K时,窗口内任意子数组中的元素之积也小于K:
class Solution {
public:int numSubarrayProductLessThanK(vector<int>& nums, int k) {if (k <= 1) {return 0;}int curProduct = 1;int left = 0;int ans = 0;for (int i = 0; i < nums.size(); ++i) {curProduct *= nums[i];while (curProduct >= k) {curProduct /= nums[left];++left;}ans += i - left + 1;}return ans;}
};
如果nums的大小为n,则此算法时间复杂度为O(n),空间复杂度为O(1)。