978. 最長湍流子陣列
阿新 • • 發佈:2021-02-17
技術標籤: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]
解釋:(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
思路
動態規劃
class Solution:
def maxTurbulenceSize(self, arr: List[int]) -> int:
n = len(arr)
dp1 = [1]*n
dp2 = [1]*n
for i in range(1, n):
if (i%2 == 0 and arr[i-1] > arr[i]) or (i%2 == 1 and arr[i-1] < arr[i]):
dp1[i] = dp1[i-1] + 1
if (i%2 == 0 and arr[i-1] <arr[i]) or (i%2 == 1 and arr[i-1] > arr[i]):
dp2[i] = dp2[i-1] + 1
return max([max(dp1) , max(dp2)])