LeetCode-Python-1343. 大小为 K 且平均值大于等于阈值的子数组数目(数组 + 滑动窗口)

 

给你一个整数数组 arr 和两个整数 k 和 threshold 。

请你返回长度为 k 且平均值大于等于 threshold 的子数组数目。

 

示例 1:

输入:arr = [2,2,2,2,5,5,5,8], k = 3, threshold = 4
输出:3
解释:子数组 [2,5,5],[5,5,5] 和 [5,5,8] 的平均值分别为 4,5 和 6 。其他长度为 3 的子数组的平均值都小于 4 (threshold 的值)。
示例 2:

输入:arr = [1,1,1,1,1], k = 1, threshold = 0
输出:5
示例 3:

输入:arr = [11,13,17,23,29,31,7,5,2,3], k = 3, threshold = 5
输出:6
解释:前 6 个长度为 3 的子数组平均值都大于 5 。注意平均值不是整数。
示例 4:

输入:arr = [7,7,7,7,7,7,7], k = 7, threshold = 7
输出:1
示例 5:

输入:arr = [4,4,4,4], k = 4, threshold = 1
输出:1
 

提示:

1 <= arr.length <= 10^5
1 <= arr[i] <= 10^4
1 <= k <= arr.length
0 <= threshold <= 10^4

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-sub-arrays-of-size-k-and-average-greater-than-or-equal-to-threshold
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:

究极Easy版滑动窗口,

把所有的长度为k的子数组找出来,再判断子数组的和是否 >= k * threshold即可。

时间复杂度:O(N)

空间复杂度:O(1),有点困惑于,for循环里range的部分是否会计算到空间复杂度里,但是转念一想改成while循环就肯定是O(1)了

class Solution(object):
    def numOfSubarrays(self, arr, k, threshold):
        """
        :type arr: List[int]
        :type k: int
        :type threshold: int
        :rtype: int
        """
        res = 0
        target_sum = k * threshold
        subs = sum(arr[0:k])
        
        for i in range(0, len(arr) - k + 1):
            if i:
                subs = subs - arr[i - 1] + arr[i + k - 1]
            if subs >= target_sum:
                res += 1
                
        return res

 

发布了722 篇原创文章 · 获赞 106 · 访问量 19万+

猜你喜欢

转载自blog.csdn.net/qq_32424059/article/details/104275031