LeetCode376. Wiggle Subsequence

题目描述：

如果连续数字之间的差严格地在正数和负数之间交替，则数字序列称为摆动序列。第一个差（如果存在的话）可能是正数或负数。少于两个元素的序列也是摆动序列。例如， [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
解释: 整个序列均为摆动序列。

示例 2:
输入: [1,17,5,10,13,15,10,5,16,8]
输出: 7
解释: 这个序列包含几个长度为 7 摆动序列，其中一个可为[1,17,10,13,10,16,8]。

示例 3:
输入: [1,2,3,4,5,6,7,8,9]
输出: 2

进阶:你能否用 O(n) 时间复杂度完成此题?

思路一（dp）：–时间O(n），空间O(n)

看到"最大子序列"这个关键词，第一个想到dp，但是怎么dp是个难点。
既是摆动序列，序列会上下起伏，

• up[i]记录上升沿时，以i结尾最大摆动序列长度；
• 类似的，down[i]记录下降沿时，以i结尾最大摆动序列长度；
• 最后返回max(up[n-1],down[n-1])

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        if(nums.size()<2)  return nums.size();
        int n=nums.size();
        vector<int> up(n,0);//up[i]上升沿时，以i结尾最大摆动序列长度
        vector<int> down(n,0);//down[i]下降沿时，以i结尾最大摆动序列长度
        up[0]=1;
        down[0]=1;
        for(int i=1;i<n;++i){
            if(nums[i]>nums[i-1]){//up
                up[i]=down[i-1]+1;
                down[i]=down[i-1];
            }
            else if(nums[i]<nums[i-1]){//down
                down[i]=up[i-1]+1;
                up[i]=up[i-1];
            }
            else if(nums[i]==nums[i-1]){//keep
                up[i]=up[i-1];
                down[i]=down[i-1];
            }
        }
        return max(up[n-1],down[n-1]);
    }
};

空间优化 --O(1)

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

思路二：（贪心）–时间O(n)，空间O(1)

看一个例子，[1,17,5,10,13,15,10,5,16,8]，当数组遍历到第四位时，即
[1,17,5,10…时，5,10,13,15形成上升序列，显然这四个数只需选一个作为摇摆序列的一部分。
选择哪个呢？若选择10，则15后面的10与其不会摇摆，故应该选大的，这样后面摇摆的概率更高！
这是贪心的思想。
再看一个图片 source of picture

从上图中也可以看出，如果选择C作为摆动序列的第二个点，则无法包括E进去。
因此本题等价于寻找数组中交替的最大最小值。

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        if(nums.size()<2)   return nums.size();
        int pre_diff=nums[1]-nums[0];
        int cnt;
        pre_diff==0?cnt=1:cnt=2;
        for(int i=2;i<nums.size();++i){
            int diff=nums[i]-nums[i-1];
            //上升序列碰到下降沿 或 下降序列碰到上升沿
            if(pre_diff>=0 && diff<0 || pre_diff<=0 && diff>0){                
                cnt++;
                pre_diff=diff;//只有摇摆时才更新pre_diff
                //如果时时更新遇到diff=0会丢失序列状态
            }
        }
        return cnt;
    }
};