Leetcode 376. 摆动序列 (动态规划到贪心优化）

采用和Leetcode 978最长湍流序列相同的动态规划方法，可以在O(n^2)的时间内解决 （原因是子序列）

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        // O(N^2)的写法
        int n = nums.size(), res = 1;
        if(n==0) return 0;
        // f[n] 表示以i结尾的上升摆动序列
        // g[n] 表示以i结尾的下降摆动序列
        vector<int> f(n,1),g(n,1);
        for(int i=1;i<n;i++){
            for(int j=0;j<i;j++){
                if(nums[i]>nums[j]) f[i] = max(f[i],g[j]+1);
                if(nums[i]<nums[j]) g[i] = max(g[i],f[j]+1);
            }
            res = max(res,max(f[i],g[i]));
        }
        return res;
    }
};

如何优化呢，要利用贪心的思想，我们把数组图示画出来发现，最长摆动序列就是端点两个值加上中间局部最大值。

建议去重以后求离散极值，向前差分，向后差分

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        nums.erase(unique(nums.begin(),nums.end()),nums.end());   // 去重
        if (nums.size()<= 2) return nums.size();
        int res = 2;
        for (int i = 1;i<nums.size()-1;i++) {
            int fordDiff = nums[i]-nums[i-1];
            int backDiff = nums[i+1] - nums[i];
            if((fordDiff<0&&backDiff>0)||(fordDiff>0&&backDiff<0)) res++;
        }
        return res;
    }
};