力扣刷题之路 978. 最长湍流子数组 Python解

当 A 的子数组 A[i], A[i+1], …, A[j] 满足下列条件时，我们称其为湍流子数组：

若 i <= k < j，当 k 为奇数时， A[k] > A[k+1]，且当 k 为偶数时，A[k] < A[k+1]；
或 若 i <= k < j，当 k 为偶数时，A[k] > A[k+1] ，且当 k 为奇数时， A[k] < A[k+1]。
也就是说，如果比较符号在子数组中的每个相邻元素对之间翻转，则该子数组是湍流子数组。

返回 A 的最大湍流子数组的长度。

示例 1：

输入：[9,4,2,10,7,8,8,1,9]
输出：5
解释：(A[1] > A[2] < A[3] > A[4] < A[5])
示例 2：

输入：[4,8,12,16]
输出：2
示例 3：

输入：[100]
输出：1

提示：

1 <= A.length <= 40000
0 <= A[i] <= 10^9

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/longest-turbulent-subarray

解：

    def maxTurbulenceSize(self, arr: List[int]) -> int:
        res = [1]
        flag = 0
        for i in range(len(arr)-1):
            # print(flag, arr[i], arr[i+1], res)
            if res[-1] == 1:
                if i:
                    if arr[i-1] < arr[i]:
                        flag = 1
                    elif arr[i-1] > arr[i]:
                        flag = 0
                    else:
                        continue
                    res[-1] += 1
                else:
                    if arr[i] < arr[i+1]:
                        flag = 1
                    elif arr[i] > arr[i+1]:
                        flag = 0
                    else:
                        continue
                    res[-1] += 1
                    continue
            if flag:
                if arr[i] > arr[i+1]:
                    res[-1] += 1
                    flag = 0
                else:
                    res.append(1)
            else:
                if arr[i] < arr[i+1]:
                    res[-1] += 1
                    flag = 1
                else:
                    res.append(1)
        return max(res)