所有元素均大于X的段数

在一个数组中，有多个段（片段），一段被定义为数组中连续的若干个元素。要求写一个程序，计算出所有元素均大于X的段数。

算法思路

我们可以采用双指针的方法，定义左边界left从头开始遍历数组，右边界right从左边界开始向后移动，直到找到一个“坏”的元素，即小于等于X的元素。此时我们就找到了一个段，我们将该段的长度（即右边界减去左边界）累加到答案中，然后将左边界left指向右边界right的下一个位置，右边界right也从left开始继续向后移动，直到找到下一个“坏”的元素。

代码实现

def count_segments(arr, X):
    left, right, cnt = 0, 0, 0
    n = len(arr)
    while left < n:
        while right < n and arr[right] > X:
            right += 1
        cnt += (right - left)
        left = right + 1
        right = left
    return cnt

测试示例

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