当前位置:   article > 正文

乘积小于 K 的子数组(滑动窗口)_滑动窗口解乘积小于 k 的子数组

滑动窗口解乘积小于 k 的子数组


简要分析,这个题目刚开始确实没想到滑动窗口,只是想到动态规划解法,时间复杂度为O(n^{2}),代码如下:

  1. class Solution {
  2. public int numSubarrayProductLessThanK(int[] nums, int k) {
  3. int n = nums.length;
  4. int ans = 0;
  5. for(int i = 0 ; i < n ; i++)
  6. {
  7. int temp = nums[i];
  8. if(temp >= k)
  9. continue;
  10. else
  11. ans++;
  12. for(int j = 1 ; j < n -i ; j++)
  13. {
  14. temp = temp * nums[i + j];
  15. if(temp < k)
  16. {
  17. ans++;
  18. }
  19. else
  20. break;
  21. }
  22. }
  23. return ans;
  24. }
  25. }

简单提交后,效果有点惨:

 几乎是马上就要超时的状态。

滑动窗口的解法,因为数组内全部都是大于0的正数,所以随着窗口的扩大,乘积肯定是越来越大的状态,不满足条件时候便缩小窗口,这么一思考,代码的基本逻辑就。

缩小窗口的代码:用ans窗口区间的乘积

  1. while(i <= j && ans >= k)
  2. {
  3. ans /= nums[i];
  4. i++;
  5. }

 总体代码如下:

  1. class Solution {
  2. public int numSubarrayProductLessThanK(int[] nums, int k) {
  3. int n = nums.length , i = 0;
  4. int ans = 1;
  5. int count = 0;
  6. for(int j = 0 ; j < n ; j++)
  7. {
  8. ans *= nums[j];
  9. while(i <= j && ans >= k)
  10. {
  11. ans /= nums[i];
  12. i++;
  13. }
  14. count += (j - i + 1);
  15. }
  16. return count;
  17. }
  18. }

时间复杂度只有O(n),

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/花生_TL007/article/detail/276926
推荐阅读
相关标签
  

闽ICP备14008679号