713. 乘积小于 K 的子数组 ●●
描述
给你一个整数数组 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. 双指针 – 滑动窗口
求连续子树组的个数,“连续子树组”很强的暗示——使用滑动窗口。本题只需记录以每个数字为右边界所形成的有效子数组的个数。
设置左右指针 l e f t 、 r i g h t left、right left、right,外循环每次移动一次右指针;
计算乘积,如果乘积过大,则保持右指针不变,内循环移动左指针,缩减窗口大小,直到满足条件;
那么该子数组中以 n u m s [ r i g h t ] nums[right] nums[right] 结尾的子子数组都满足条件,个数为 r i g h t − l e f t + 1 right - left + 1 right−left+1。
- 时间复杂度: O ( n ) O(n) O(n),其中 n 是数组 nums 的长度。两个指针移动的次数都不超过 n。
- 空间复杂度: O ( 1 ) O(1) O(1)。
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
int n = nums.size();
int left = 0, multi = 1, ans = 0;
for(int right = 0; right < n; ++right){
multi *= nums[right]; // 右边界右移
while(multi >= k && left <= right){ // left不能大于right
multi /= nums[left];
++left; // 乘积过大时,右移left指针缩小窗口
}
ans += right - left + 1; // 以nums[right]结尾,满足条件的子数组的长度
}
return ans;
}
};