LeetCode 978. 最长湍流子数组(DP)

1. 题目

当 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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/longest-turbulent-subarray
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

  • 两个dp数组,dp0,表示以下降结束,dp1表示以上升结束
class Solution {	// c++
public:
    int maxTurbulenceSize(vector<int>& A) {
    	int i, n = A.size(), maxlen = 1;
    	vector<int> dp0(n,1), dp1(n,1);
    	//dp0,表示以下降结束,dp1表示以上升结束
    	// 到位置 i 处的最长长度
    	for(i = 1; i < n; i++)
    	{
    		if(A[i-1] > A[i])//需要前面是上升,现在是下降
    			dp0[i] = dp1[i-1] + 1;
    		else if(A[i-1] < A[i])
    			dp1[i] = dp0[i-1] + 1;
    		maxlen = max(maxlen, max(dp0[i],dp1[i]));
    	}
    	return maxlen;
    }
};

164 ms 41.2 MB

class Solution:	# py3
    def maxTurbulenceSize(self, A: List[int]) -> int:
        n = len(A)
        maxlen = 1
        dp0 = [1]*n
        dp1 = [1]*n
        for i in range(1,n):
            if A[i-1] > A[i]:
                dp0[i] = dp1[i-1]+1
            elif A[i-1] < A[i]:
                dp1[i] = dp0[i-1]+1
            maxlen = max(maxlen, max(dp0[i], dp1[i]))
        return maxlen

876 ms 17.6 MB

猜你喜欢

转载自blog.csdn.net/qq_21201267/article/details/106590866