376. 摆动序列

Posted 2021-03-08

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

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

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

代码实现:

---

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        if(nums.size() < 2)
        return nums.size();
        const int begin = 0;
        const int up = 1;
        const int down = 2;
        int state = begin;
        int max_length = 1;
        for(int i = 1; i < nums.size(); i++)
        {
            switch(state)
            {
            case begin:
            {
                if(nums[i -1] < nums[i])
                {state = up;
                max_length++;
                }
                if(nums[i-1] > nums[i])
                {
                    state = down;
                    max_length++;
                }
            }
            break;
            case up:
            if(nums[i-1] > nums[i])
            {
                state = down;
                max_length++;
            }
            break;
            case down:
            if(nums[i-1] < nums[i])
            {
                state = up;
                max_length++;
            }
            }
        }
        return max_length;
    }
};