LeetCode---376. 摆动序列（贪心or动态规划）

方法一（贪心）

示例代码

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        int len=nums.size();
        if(len<=1){
            return len;
        }
        //摆动序列个数等于峰值的个数加一
        int curdiff=0,prediff=0,result=1;
        for(int i=0;i+1<len;i++){
            curdiff=nums[i+1]-nums[i];
            if((prediff>=0&&curdiff<0)||(prediff<=0&&curdiff>0)){
                result++;
                //只有当峰值发生变化的时候才更新前一个的方向
                prediff=curdiff;
            }
        }
        return result;
    }
};

效果展示

方法二（动态规划）

示例代码

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        int len=nums.size();
        vector<int> up(len),down(len);
        //数组长度为1的时候认为摆动序列为1
        up[0]=1,down[0]=1;
        //从前往后不断地更新摆动序列
        for(int i=1;i<len;i++){
            //此数字比前一个数字大的时候更新up
            if(nums[i]>nums[i-1]){
                up[i]=max(up[i-1],down[i-1]+1);
            }
            //此数字比前一个数字小的时候更新down
            else if(nums[i]<nums[i-1]){
                down[i]=max(down[i-1],up[i-1]+1);
            }
            //此数字和前一个数字大小相同时down和up都不会发生变化
            else{
                up[i]=up[i-1];
                down[i]=down[i-1];
            }
        }
        return max(up[len-1],down[len-1]);

    }
};

效果展示