leetcode力扣刷题打卡
题目:1. 两数之和
描述:给你一个整数数组 nums 和一个整数 k ,请你返回子数组内所有元素的乘积严格小于 k 的连续子数组的数目。
解题思路
1、双指针;
2、左指针l 不动,右指针一直r++,直到前缀乘积mul >= k,此时用一个while循环,让左指针不断左移。
3、该[l, r]区间内的子数组数量就是r - l + 1;
4、第二个while循环一定要加上判断l <= r, 因为可能出现数组为[1, 2, 3]和k == 0的情况,永远也到不了mul < k的情况,这时候l就会一直往后走,进入死循环。
原代码##
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
int n = nums.size();
int l = 0, r = 0, ans = 0;
int mul = 1;
for (r = 0; r < n; ++r) {
mul *= nums[r];
while (l <= r && mul >= k) {
mul /= nums[l];
l++;
}
ans += r - l + 1;
}
return ans;
}
};
或者
class Solution {
public:
int numSubarrayProductLessThanK(vector<int>& nums, int k) {
int n = nums.size();
int l = 0, r = 0, ans = 0, mul = 1;
while (r < n) {
mul *= nums[r];
while (mul >= k && l <= r) {
mul /= nums[l];
l++;
}
ans += r - l + 1;
r++;
}
return ans;
}
};