LeetCode 376. Wiggle Subsequence 非常简单的解法

问题

如果连续数之间的差值严格地交替出现在正数和负数之间，那么这个数列就称为摆动数列。第一个区别(如果存在的话)可能是积极的，也可能是消极的。一个少于两个元素的序列通常是一个摆动序列。例如，[1,7,4,9,2,5]是一个摆动序列，因为差异(6，-3,5，-7,3)交替为正和负。相反，[1,4,7,2,5]和[1,7,4,5,5]不是摆动序列，第一个是因为它的前两个差异是正的，第二个是因为它的最后一个差异是零。
给定一个整数序列，返回一个摆动序列的最长子序列的长度。子序列是通过从原始序列中删除一些元素(最终也为零)来获得的，而其余的元素则是按照原始顺序保留的。

输入: [1,17,5,10,13,15,10,5,16,8]
输出: 7
解释: 有多个子序列满足要求. 其中一个是 [1,17,10,13,10,16,8].

解析

这题看似放在dp里面，其实有一个更简单的解法：子序列长度能增加的条件当前增长的方向和之前增长的方向相反，也就是说之前的符号是+，那么当前的符号是-；或者相反。而且每次增加，符号必须替换成当前的符号以供下一次比较，以此类推，可以写出代码：

Java代码

    public int wiggleMaxLength(int[] nums) {
        if(nums.length<2) return nums.length;
        int flag = nums[1]>nums[0]?1:-1;
        if(nums[1]==nums[0]) flag =0;
        int ans = nums[1]-nums[0]!=0?2:1;;
        for(int i=1;i<nums.length-1;i++){
            if(flag>=0&&nums[i+1]<nums[i]){
                ans++;
                flag = -1;
            }
            else if(flag<=0&&nums[i+1]>nums[i]){
                ans++;
                flag = 1;
            }
        }
        return ans;
    }