【LeetCode】376. 摆动序列 结题报告 (C++)

原题地址：https://leetcode-cn.com/problems/wiggle-subsequence/submissions/

题目描述：

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

示例 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) 时间复杂度完成此题?

 

解题方案：

使用动态规划和贪心算法解题，解题的思路不太难想，采用两个数组作为动态规划数组。

代码：

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

给出时间最短的算法：

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        if(nums.size()<2)
            return nums.size();
        
        static const int BEGIN=0;
        static const int UP=1;
        static 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]>nums[i-1])
                    {
                        STATE=UP;
                        max_length++;
                    }
                    else if(nums[i]<nums[i-1])
                    {
                        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++;
                    }
                    break;
            }
        }
        return max_length;
    }
};