給你一個整數數組 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)。