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;
    }
};