leetcode 978.最长湍流子数组

题干

当 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

题解

朴素的dp * 因为湍流子数组的判断条件把大于小于号调转过来也正确，所以遍历了两次 * 设dp[i]为以第i个元素为结尾的最长湍流子数组的长度： 考虑状态转移方程： dp[i] = dp[i-1] + 1; 当且仅当i为偶数，则A[i]<A[i-1],i为奇数A[i]>A[i-1] else dp[i] = 1; 考虑边界条件： dp[0] = 1;

class Solution { public: int maxTurbulenceSize(vector<int>& A) { int n = A.size(); int dp[n+5]; int ans = INT_MIN; dp[0] = 1; for(int i=1;i<n;i++){ if(((i%2==0) && (A[i]<A[i-1])) || ((i%2==1) && (A[i]>A[i-1]))) dp[i] = dp[i-1] + 1; else dp[i] = 1; } for(int i=0;i<n;i++) ans = max(ans,dp[i]); for(int i=1;i<n;i++){ if(((i%2==0) && (A[i]>A[i-1])) || ((i%2==1) && (A[i]<A[i-1]))) dp[i] = dp[i-1] + 1; else dp[i] = 1; } for(int i=0;i<n;i++) ans = max(ans,dp[i]); return ans; } };

整理一下看起来美观一点

class Solution { public: int maxTurbulenceSize(vector<int>& A) { int n = A.size(); int dp[n+5][2]; int ans = INT_MIN; dp[0][0] = dp[0][1] = 1; for(int i=1;i<n;i++){ if(((i%2==0) && (A[i]<A[i-1])) || ((i%2==1) && (A[i]>A[i-1]))) dp[i][0] = dp[i-1][0] + 1; else dp[i][0] = 1; if(((i%2==0) && (A[i]>A[i-1])) || ((i%2==1) && (A[i]<A[i-1]))) dp[i][1] = dp[i-1][1] + 1; else dp[i][1] = 1; } for(int i=0;i<n;i++) ans = max(max(ans,dp[i][0]),dp[i][1]); return ans; } };