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

二、解题思路

        通过分析题目可以得到的结论为：如果当前元素和其前后元素的大小关系交替变换，则可以组成湍流子数组。假如大小关系用1、0、-1来表示的话，湍流数组中元素的关系应为：1、-1、1、-1......这种周期性变化。根据这一原理，可以按照如下步骤进行解题：

（1）比较任意元素和其前后元素之间的关系，如果前后关系相乘为-1，则湍流子数组长度+1；否则记录下目前湍流子数组长度的最大值，并以当前元素为湍流子数组的起点，重新记录湍流子数组的长度。

（2）处理边界头，因为按照题目要求数组长度至少为1，所以湍流子数组的长度同样至少为1，并从第二个元素开始进行遍历。

（3）处理边界尾，对于最后一个元素，只需判断它与前一个元素不等即可。

三、解题代码

class Solution {
public:
    int maxTurbulenceSize(vector<int>& arr) {
        int n = arr.size();
        int ans = 1; //长度至少为1
        int pos = 0; //记录湍流子数组的起点
        for(int i = 1; i < n; ++i)
        {
            //当前项与前一项之间的关系
            int c = compare(arr[i], arr[i-1]); 
            //处理最后一项以及前后大小关系没有交替变化的情况
            if(i == n-1 || c * compare(arr[i+1], arr[i]) != -1)
            {
                //当前项与前一项不等时
                if(c)
                    ans = max(ans, i - pos + 1);
                //更新湍流子数组的起点
                pos = i;
            }
        }
        return ans;
    }

    int compare(int a, int b)
    {
        if(a - b > 0)
            return 1;
        else if(a - b < 0)
            return -1;
        else
            return 0;
    }

};

四、解题结果