Leetcode 1343:大小为K且平均值大于等于阈值的子数组数目(超详细的解法!!!)
in leetcode with 0 comment

Leetcode 1343:大小为K且平均值大于等于阈值的子数组数目(超详细的解法!!!)

in leetcode with 0 comment

给你一个整数数组 arr 和两个整数 kthreshold

请你返回长度为 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

提示:

解题思路

首先不难想到前缀和的思路,简单提一下,例如想要计算区间arr[i:j]的和,那么可以通过sum(arr[:j])-sum(arr[:i-1]),而区间arr[:k]的和可以通过不断向后累加得到。

class Solution:
    def numOfSubarrays(self, arr: List[int], k: int, threshold: int) -> int:
        n = len(arr)
        pre = [0] * (n  + 1)
        for i in range(1, n + 1):
            pre[i] = pre[i - 1] + arr[i - 1]
        
        res = 0
        for i in range(k, n + 1):
            if pre[i] - pre[i - k] >= threshold * k:
                res += 1
        return res

对于这种区间问题还可以使用滑动窗口,例如

2  2  2  2  5  5  5  8
l        r

每次将(l, r]这个区间向右移动一步,而区间中的和变化了arr[r]-arr[l]

class Solution:
    def numOfSubarrays(self, arr: List[int], k: int, threshold: int) -> int:
        s, res, t = sum(arr[:k - 1]), 0, k * threshold
        
        for l in range(len(arr) - k + 1):
            s += arr[l + k - 1]
            if s >= t:
                res += 1
            s -= arr[l]
        return res

我将该问题的其他语言版本添加到了我的GitHub Leetcode

如有问题,希望大家指出!!!

「如果我的文章对你有很大帮助,那么不妨~!」

coordinate

谢谢老板O(∩_∩)O~

使用微信扫描二维码完成支付

Responses