【Leetcode】713. Subarray Product Less Than K

题目地址:

https://leetcode.com/problems/subarray-product-less-than-k/

给定一个正整数长 n n n的数组 A A A,再给定一个非负整数 k k k,问 A A A中有多少个子数组的乘积是小于 k k k的。

思路是双指针。开左右两个指针 j j j i i i,再用一个变量维护 ∏ A [ j : i ] \prod A[j:i] A[j:i],枚举子数组右端点,如果乘积小于 k k k,那么以 A [ i ] A[i] A[i]为右端点的满足条件的子数组的个数就是 i − j + 1 i-j+1 ij+1;否则右移左指针直到乘积小于 k k k。注意,每次枚举完右端点之后,左指针不必回退,因为这只会导致乘积更大。代码如下:

public class Solution {
    
    
    public int numSubarrayProductLessThanK(int[] nums, int k) {
    
    
        int res = 0, prod = 1;
        for (int i = 0, j = 0; i < nums.length; i++) {
    
    
            prod *= nums[i];
            while (j <= i && prod >= k) {
    
    
                prod /= nums[j];
                j++;
            }
            
            res += i - j + 1;
        }
        
        return res;
    }
}

时间复杂度 O ( n ) O(n) O(n),空间 O ( 1 ) O(1) O(1)

猜你喜欢

转载自blog.csdn.net/qq_46105170/article/details/112607496
今日推荐