计算所有元素大于 K 的子数组

本文将介绍如何编写一个程序，计算给定数组中所有元素大于 K 的子数组数量。我们将使用Python语言实现该功能。

思路

我们可以遍历给定数组，对每个数进行判断是否大于K，并记录下标，然后使用数学公式计算子数组数目。

具体操作如下：

1. 初始化变量：count=0，index=None。
2. 遍历数组，对每个元素判断是否大于K。
3. 如果元素大于K，并且上一个元素小于等于K，则更新index为当前下标。
4. 如果index不为空，则计算以index为起点的子数组个数。子数组个数为当前下标减去index加1，加到count中。
5. 结束遍历，返回count。

代码

以下是完整的Python代码实现（其中输入的数组为list类型）：

def count_subarrays(arr, k):
    """
    计算所有元素大于 K 的子数组数量
    :param arr: 输入的数组
    :param k: 给定值 K
    :return: 计算出的子数组数量
    """
    count = 0
    index = None
    for i, num in enumerate(arr):
        if num > k:
            if i > 0 and arr[i-1] <= k:
                index = i
            if index is not None:
                count += i - index + 1
        else:
            index = None
    return count

测试

以下是一些测试用例，可以验证程序的正确性。

arr1 = [2, 4, 6, 8, 10]
k1 = 5
res1 = count_subarrays(arr1, k1)
assert res1 == 10

arr2 = [1, 2, 3, 4, 5]
k2 = 3
res2 = count_subarrays(arr2, k2)
assert res2 == 9

arr3 = [3, 2, 1]
k3 = 1
res3 = count_subarrays(arr3, k3)
assert res3 == 3

print("测试通过！")

结语

使用上述方法可以快速计算给定数组中所有元素大于 K 的子数组个数。开发者可以根据需要进行优化，例如使用滑动窗口等算法来优化性能。