leetcode算法

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

提示:

分析:

这题数组长度有些打,先尝试暴力,看看能不能通过。本题暴力是可以通过的

暴力遍历

 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
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

上一篇下一篇

猜你喜欢

热点阅读