376. 摆动序列

难度中等260收藏分享切换为英文接收动态反馈

如果连续数字之间的差严格地在正数和负数之间交替，则数字序列称为摆动序列。第一个差（如果存在的话）可能是正数或负数。少于两个元素的序列也是摆动序列。

例如， [1,7,4,9,2,5] 是一个摆动序列，因为差值 (6,-3,5,-7,3) 是正负交替出现的。相反, [1,4,7,2,5] 和 [1,7,4,5,5] 不是摆动序列，第一个序列是因为它的前两个差值都是正数，第二个序列是因为它的最后一个差值为零。

给定一个整数序列，返回作为摆动序列的最长子序列的长度。 通过从原始序列中删除一些（也可以不删除）元素来获得子序列，剩下的元素保持其原始顺序。

示例 1:

输入: [1,7,4,9,2,5]
输出: 6 
解释: 整个序列均为摆动序列。

 

摆动序列中的数一定是上升或者下降状态

up设定为nums[0:i]中摆动序列最后一个元素nums[i]是上升状态的最长长度

down设定为nums[0:i]中摆动序列最后一个元素nums[i]是下降状态的最长长度

如果nums[i+1]>nums[i]，那么nums[i+1]肯定是上升，nums[:i]是下降状态，更新

up = down + 1

如果nums[i+1]<nums[i]，那么nums[i+1]肯定是下降，更新down = up + 1

 

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        int n = nums.size();
        if(!n) return 0;
        int up = 1, down = 1;
        for(int i = 1;i < n;i++)
        {
            if(nums[i] > nums[i-1])
                up = down + 1;
            else if(nums[i] < nums[i-1])
                down = up + 1;
        }
        return max(up, down);
    }
};