LeetCode376. 摆动序列

题目

分析

摘自Carl大佬总结

局部最优：删除单调坡度上的节点（不包括单调坡度两端的节点），那么这个坡度就可以有两个局部峰值。实际操作上，其实连删除的操作都不用做，因为题目要求的是最长摆动子序列的长度，所以只需要统计数组的峰值数量就可以了（相当于是删除单一坡度上的节点，然后统计长度）但题目只要求返回i长度，所以并不需要真的删除。最后统计局部峰值即可。

关键是统计峰值也不好统计有一定的技巧。下面的错误代码就是自己在统计峰值时的错误

错误代码（主要思想是去除峰底和峰顶之间坡上的元素）

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        int res = nums.size();
        //边界处理
        if(nums.size() == 0 || nums.size() == 1) return nums.size(); //没有元素
        if(nums.size() == 2 && nums[0] == nums[1]) return nums.size()-1; //两个相同元素
        if(nums.size() == 3 && nums[0] == nums[1] && nums[1] == nums[2]) return 1;
        
        for(int i = 1;i <nums.size()-1;i++){  //检查从第二个元素到倒数第二个元素
            if(nums[i] > nums[i-1] && nums[i+1] < nums[i]) {cout<<nums[i]<<" ";continue;}
            if(nums[i] < nums[i-1] && nums[i+1] > nums[i]) {cout<<nums[i]<<" ";continue;}
            if(i > 1 && nums[i] == nums[i-1] && nums[i] == nums[i+1]) {cout<<nums[i];continue;}
            res--;
        }
        return res;
    }
};

上面代码主要总差1，是因为连续三个相同值、连续两个相同值出错，总之讨论繁琐。从总个数中删除非极值点不好做。

正确代码

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        if(nums.size() <= 1) return nums.size();
        int res = 1;//初始为1，默认最右段有峰值
        int cur = 0;//当前对的差值
        int pre = 0;//上一对的差值,前置0
        for(int i = 1;i < nums.size();i++){
            cur = nums[i] - nums[i-1];
            if((cur > 0 && pre <= 0) || (cur < 0 && pre >= 0)){
                res++;
                pre = cur;
            }
        }
        return res;
    }
};