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]) < 0j向右移动。
单独处理了数组中只有一个元素的情况!

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