首页 > 编程语言 > 详细

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

时间:2021-04-16 12:58:13      阅读:26      评论:0      收藏:0      [点我收藏+]

难度 medium
给你一个整数数组 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

解题思路:这里主要就是用一个数组序列来记录原数组的累加值,然后根据这个累加值序列中索引相差k的两个元素的差值,也就是中间k个元素的累加值,来与k*threshold进行比较。这里需要注意的地方是,累加值序列比原来的序列要多一个元素,主要是考虑到第k个元素时(也就是j=k),sum[j]-sum[j-k]需要用到0索引,这个零索引表示没有任何元素相加的累加值,也就是我们需要考虑设定的一个初始条件。

代码 t60 s9 java

class Solution {
    public int numOfSubarrays(int[] arr, int k, int threshold) {
        int len = arr.length;
        int[] acc = new int[len+1];
        for(int i=1; i<=len; i++){
            acc[i] = acc[i-1] + arr[i-1];
        }
        int res = 0;
        for(int i=k; i<=len; i++){
            if(acc[i]-acc[i-k] >= k*threshold){
                res++;
            }
        }
        return res;
    }
}

参考资料

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

原文:https://www.cnblogs.com/zhengxch/p/14665523.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!