LeetCode 978. 最长湍流子数组（C++、python）

最新推荐文章于 2021-02-08 23:28:50 发布

我很忙2010

最新推荐文章于 2021-02-08 23:28:50 发布

阅读量324

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/qq_27060423/article/details/86600056

版权

LeetCode 专栏收录该内容

713 篇文章 10 订阅

订阅专栏

当 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

C++

class Solution {
public:
	int maxTurbulenceSize(vector<int>& A)
	{
		int n = A.size();
		int res = 1;
		int tmp = 1;
		int flag;
		for (int i = 1; i<n; i++)
		{
			if (1 == i)
			{
				if (A[i] > A[i - 1])
				{
					tmp++;
					flag = 1;
				}
				else if (A[i] < A[i - 1])
				{
					tmp++;
					flag = -1;
				}
				else
				{
					flag = 0;
				}
				res = max(tmp, res);
			}
			else
			{
				if (A[i] > A[i - 1])
				{
					if (-1 == flag)
					{
						tmp += 1;
						res = max(tmp, res);
					}
					else
					{
						tmp = 2;
					}
					flag = 1;
				}
				else if (A[i] < A[i - 1])
				{
					if (1 == flag)
					{
						tmp += 1;
						res = max(tmp, res);
					}
					else
					{
						tmp = 2;
					}
					flag = -1;
				}
				else
				{
					tmp = 1;
					flag = 0;
				}
			}
		}
		return res;
	}
};

python

class Solution:
    def maxTurbulenceSize(self, A):
        """
        :type A: List[int]
        :rtype: int
        """
        n=len(A)
        res=1
        tmp=1
        flag=1
        for i in range(1,n):
            if 1==i:
                if A[i]>A[i-1]:
                    tmp+=1
                    flag=1
                elif A[i]<A[i-1]:
                    tmp+=1
                    flag=-1
                else:
                    flag=0
                res=max(res,tmp)
            else:
                if A[i]>A[i-1]:
                    if -1==flag:
                        tmp+=1
                        res=max(res,tmp)
                    else:
                        tmp=2
                    flag=1
                elif A[i]<A[i-1]:
                    if 1==flag:
                        tmp+=1
                        res=max(res,tmp)
                    else:
                        tmp=2
                    flag=-1
                else:
                    tmp=1
                    flag=0
        return res