1343. 大小为 K 且平均值大于等于阈值的子数组数目 [ar

【Description】
给你一个整数数组 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

【Idea】
两个tips：

1. 子数组是指原数组中连续的一个元素子集，不能任意元素组合；
2. 接上，既然是保留初始元素顺序，那么为减少时间复杂度，就要用到滑窗trick来优化，不能暴力解。

【Solution】

class Solution:
    def numOfSubarrays(self, arr: List[int], k: int, threshold: int) -> int:
        length = len(arr)
        if length < k or (length == k and sum(arr) < k*threshold):
            return 0
        s = sum(arr[:k])
        if s >= k * threshold:
            res = 1
        else:
            res = 0
        for left in range(1, length-k+1):
            right = left + k - 1
            s = s - arr[left-1] + arr[right]
            if s >= k*threshold:
                res += 1
        return res

屏幕快照 2020-06-22 上午12.03.06.png