算法-Wiggle Subsequence-摆动序列

1、问题描述

如果连续数字之间的差严格地在正数和负数之间交替，则数字序列称为摆动序列。第一个差（如果存在的话）可能是正数或负数。少于两个元素的序列也是摆动序列。
    例如， [1,7,4,9,2,5] 是一个摆动序列，因为差值 (6,-3,5,-7,3) 是正负交替出现的。相反, [1,4,7,2,5] 和 [1,7,4,5,5] 不是摆动序列，第一个序列是因为它的前两个差值都是正数，第二个序列是因为它的最后一个差值为零。

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

示例 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]。

2、题目分析

• 刚开始理解错了，没有看到可以直接删除，以为要取出最大连续子序列，用了动态规划,第二个用例不通过才发现

  class Solution {
      //dp[i]以i结尾的最长摆动序列的大小
      //dp[0]=1 dp[1]=2 dp[2]=2
      //dp[i] = 1 ? dp[i - 1] + 1
      //return max
      public int wiggleMaxLength(int[] nums) {
          if(nums == null) return 0;
          int n = nums.length;
          if(n <= 2) return n;
          int cur = 0,pre = 0,max = 0;
          int[] dp = new int[n];
          dp[0] = 1;dp[1] = 2;//可以优化常数空间
          for(int i = 2;i < n;i++){
              pre = nums[i - 1] - nums[i - 2];
              cur = nums[i] - nums[i - 1];
              dp[i] = (cur * pre < 0) ? (dp[i - 1] + 1) : 1;
              max = Math.max(max,dp[i]);
          }
          return max;
      }
  }
  

• 可以删除子序列，怎么做呢？

  • 用up[i]表示最后两个数字是递增的最长摆动序列（0-i)
  • 用down[i]表示最后两个数字为递减的最长摆动序列（0-i）
  • 可以看到最后两个上升的要衔接之前的最长下降字符 up[i] = down[i - 1] + 1;
  • 同理 down[i] = up[i - 1] + 1;
  • 只跟过去的状态有关，则用两个变量替代dp数组即可

3、代码实现

class Solution {
    public int wiggleMaxLength(int[] nums) {
        int n = nums.length;
        if(n < 2) return n;
        int up = 1,down = 1;
        for(int i = 1;i < n;i++){
            if(nums[i] > nums[i - 1]) up = down + 1;
            else if(nums[i] < nums[i - 1]) down = up + 1;
        }
        return Math.max(up,down);
    }
}

4、复杂度分析

• 时间复杂度：O(N)
• 空间复杂度：O(1)