小于两个相邻元素的数组元素的数量

在一个整数数组中，如果某个元素小于相邻的两个元素，那么我们就称这个元素满足条件。本文介绍如何统计一个数组中满足条件的元素个数。

算法一

遍历数组，对于每个位置，判断它是否满足条件。时间复杂度为 O(n)。

def count(nums):
    result = 0
    for i in range(1, len(nums) - 1):
        if nums[i] < nums[i - 1] and nums[i] < nums[i + 1]:
            result += 1
    return result

算法二

该算法利用了滑动窗口的思想，可以将时间复杂度降为 O(1)。

def count(nums):
    result = 0
    i, j = 1, 2
    while j < len(nums):
        if nums[j] > nums[i] and nums[i] < nums[i - 1]:
            result += j - i - 1
            i = j - 1
        else:
            i += 1
        j += 1
    return result

测试代码

assert count([1, 2, 3, 4, 5]) == 0
assert count([5, 4, 3, 2, 1]) == 0
assert count([1, 3, 2, 1, 2, 3, 1]) == 2
assert count([1, 2, 1, 2, 1, 2, 1]) == 3

总结

算法一比较简单，但时间复杂度比较高，适用于数据规模较小的情况。算法二的时间复杂度比较低，适用于数据规模较大的情况。