leetcode 剑指 Offer II 009. 乘积小于 K 的子数组

最新推荐文章于 2024-09-30 11:37:22 发布

dengjiayue

最新推荐文章于 2024-09-30 11:37:22 发布

阅读量882

点赞数

文章标签：后端

本文链接：https://blog.csdn.net/dengjiayue/article/details/131267020

版权

文章介绍了如何使用双指针滑动窗口算法解决LeetCode上的一个问题，即在给定的正整数数组nums中找出乘积小于k的连续子数组的个数。方法是维护一个乘积变量product和两个指针left和right，动态调整窗口大小，计算满足条件的子数组数量。

摘要由CSDN通过智能技术生成

leetcode 剑指 Offer II 009. 乘积小于 K 的子数组

题目描述

剑指 Offer II 009. 乘积小于 K 的子数组

给定一个正整数数组 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 * 104 1 <= nums[i] <= 1000 0 <= k <= 106

解题思路

法1

双指针滑窗:

滑动窗口的思想。左指针 left 和右指针 right 分别表示滑动窗口的左右边界。我们维护一个变量 product 来记录窗口内元素的乘积。

在每次右移右指针 right 时，将当前元素 num 乘到 product 中，并检查乘积是否大于等于 k。

如果大于等于 k，则将左指针 left 右移，同时除去窗口最左侧的元素，直到乘积小于 k。

每次右移右指针时，都可以得到以当前右指针所指元素为结尾的满足条件的子数组数量，即 right - left + 1，并累加到结果变量 count 中。

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

执行结果

法1

func numSubarrayProductLessThanK(nums []int, k int) int {
 if k <= 1 {
  return 0
 }

 count := 0
 product := 1
 left := 0

 for right, num := range nums {
  product *= num

  for product >= k {
   product /= nums[left]
   left++
  }

  count += right - left + 1
 }

 return count
}