LeetCode刷题记录--1343. 大小为 K 且平均值大于等于阈值的子数组数目

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

审题：

1、输入为整数数组，长度为k子数组元素平均值>=threshold

2、输出满足1的子数组个数

3、数组长度[1, 10^5],元素大小[1, 10^4]，k大小[1, 数组长度]，threshold大小范围[0, 10^4]

 

思路：

1、将条件转换为长度为k的子数组和>= K * threshold

2、由1能想到使用前缀和减少运算

3、使用滑窗求满足条件的子数组个数

 

int numOfSubarrays(int* arr, int arrSize, int k, int threshold){
    int target = k * threshold;
    int *sum = (int *)malloc((arrSize + 1)* sizeof(int));
    memset(sum, 0, (arrSize + 1)* sizeof(int));
    int i;
    // 求各元素的前缀和
    for (i = 1; i <= arrSize; i++) {
        sum[i] = sum[i - 1] + arr[i - 1];
    }
    int count = 0;
    // 使用滑窗方法，判断元素前缀和判断距离为k的元素的平均值是否大于等于threshold
    for (i = k; i <= arrSize; i++) {
        int temp = sum[i] - sum[i - k];
        if (temp >= target) {
            count++;
        }
    }
    return count;
}