713. 乘积小于 K 的子数组
2022-05-05 本文已影响0人
刘翊扬
给你一个整数数组 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
- 1 <= nums[i] <= 1000
- 0 <= k <= 10^6
分析:
这题数组长度有些打,先尝试暴力,看看能不能通过。本题暴力是可以通过的
暴力遍历
public int numSubarrayProductLessThanK(int[] nums, int k) {
int ans = 0, n = nums.length;
for (int i = 0; i < n; i++) {
int tmp = nums[i];
if (tmp >= k) {
continue;
}
ans++;
for (int j = i + 1; j < n; j++) {
tmp *= nums[j];
if (tmp >= k) {
break;
}
ans++;
}
}
return ans;
}
双指针
/**
* 双指针法,如果一个子串的乘积小于k,那么他的每个子集都小于k,而一个长度为n的数组,他的所有连续子串数量是1+2+...n,但是会和前面的重复。
* 比如例子中[10, 5, 2, 6],第一个满足条件的子串是[10],第二个满足的是[10, 5],但是第二个数组的子集[10]和前面的已经重复了,
* 因此我们只需要计算包含最右边的数字的子串数量,就不会重复了,也就是在计算[10, 5]这个数组的子串是,只加入[5]和[10, 5],而不加入[10],
* 这部分的子串数量刚好是r - l + 1
* @param nums
* @param k
* @return
*/
public int numSubarrayProductLessThanK(int[] nums, int k) {
int ans = 0, n = nums.length;
int tmp = 1, i = 0;
for (int j = 0; j < n; j++) {
tmp *= nums[j];
while (i <= j && tmp >= k) {
tmp /= nums[i];
i++;
}
ans += j - i + 1;
}
return ans;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/subarray-product-less-than-k
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。