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
}
执行结果： 通过 显示详情 查看示例代码 添加备注
执行用时： 60 ms , 在所有 Go 提交中击败了 98.73% 的用户 内存消耗： 7 MB , 在所有 Go 提交中击败了 90.25% 的用户 通过测试用例： 95 / 95 炫耀一下:
本文由 mdnice 多平台发布