Leetcode 978 Longest Turbulent Subarray

发表于 |

当 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. 1 <= A.length <= 40000
  2. 0 <= A[i] <= 10^9

题意分析：
找一个数组的子串，使得其满足任意 (A[i]-A[i+1]) * (A[i-1]-A[i]) < 0

思路分析：
这里采用双指针法，用j-i+1代表子串的长度，当满足(A[j-1]-A[j])*(A[j]-A[j+1]) < 0时j向右移动。
单独处理了数组中只有一个元素的情况！

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
class Solution(object):
    def maxTurbulenceSize(self, A):
        if len(A) <= 2: return len(A)
        # 全是相同元素
        if A.count(A[0]) == len(A):
            return 1
        i,j = 0,1
        res = 1
        while j < len(A)-1:
            if (A[j-1]-A[j])*(A[j]-A[j+1]) < 0:
                j += 1
            else:
                res = max(res, j-i+1)
                i,j = j, j+1
        return max(res, j-i+1)

# 85 / 85 test cases passed.
# Status: Accepted
# Runtime: 108 ms