门| GATE CS Mock 2018 |设置 2 |问题 12

本题是 GATE CS Mock 2018 |设置 2 |问题 12，是一道典型的数组问题，需求如下：

有一个长度为 n 的整数数组 a，每个元素都是 0 或 1。

你需要找到最长的连续子数组，其中只包含 0 或 1，并返回其长度。

如数组 a = [0, 1, 0, 0, 1, 1, 0, 1, 1] 中最长的连续子数组为 [0, 1, 0, 0, 1, 1]，其长度为 6。

算法思路

可以使用指针从头遍历数组，记录当前最长的连续子数组的长度，并使用变量 cnt 记录当前子数组中 0 和 1 的数量，如果 cnt 大于 1，则将左指针右移直到 cnt 减少至 1，直至扫描完整个数组。

具体实现如下：

def max_length_subarray(a):
    left = 0
    cnt = 0
    res = 0
    for i in range(len(a)):
        if a[i] == 0:
            cnt += 1
        if cnt > 1:
            if a[left] == 0:
                cnt -= 1
            left += 1
        res = max(res, i - left + 1)
    return res

时间复杂度

遍历一遍整个数组，时间复杂度为 $O(n)$。

空间复杂度

只使用了常数级别的额外空间，空间复杂度为 $O(1)$。

测试样例

assert max_length_subarray([0, 1, 0, 0, 1, 1, 0, 1, 1]) == 6
assert max_length_subarray([0, 0, 0, 0]) == 4
assert max_length_subarray([1, 1, 1, 1]) == 4
assert max_length_subarray([1, 0, 1, 0, 1, 0]) == 1

以上为本题的算法思路、时间复杂度、空间复杂度和测试样例，可以轻松通过该题。