大小为 K 且平均值大于等于阈值的子数组数目Java
文章发布较早,内容可能过时,阅读注意甄别。
# 题目
给你一个整数数组 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 = [11,13,17,23,29,31,7,5,2,3], k = 3, threshold = 5
输出:6
解释:前 6 个长度为 3 的子数组平均值都大于 5 。注意平均值不是整数。
提示:
- 1 <= arr.length <= 105
- 1 <= arr[i] <= 104
- 1 <= k <= arr.length
- 0 <= threshold <= 104
# 思路
1、前缀和数组快速计算子数组平均值 2、枚举每个子数组的左端点找到符合条件的子数组
# 解法
class Solution {
public int numOfSubarrays(int[] arr, int k, int threshold) {
int res = 0;
int[] prefixSum = new int[arr.length + 1];
for (int i = 1; i < prefixSum.length; i++) {
prefixSum[i] = prefixSum[i - 1] + arr[i - 1];
}
for (int i = 0; i + k <= arr.length; i++) {
if ((prefixSum[i + k] - prefixSum[i]) / k >= threshold) {
res++;
}
}
return res;
}
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# 总结
- 分析出几种情况,然后分别对各个情况实现