想要精通算法和SQL的成长之路 - 摆动序列

前言

想要精通算法和SQL的成长之路 - 系列导航

一. 摆动序列

原题链接

如果连续数字之间的差严格地在正数和负数之间交替，则数字序列称为摆动序列 。第一个差（如果存在的话）可能是正数或负数。仅有一个元素或者含两个不等元素的序列也视作摆动序列。例如：

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

子序列可以通过从原始序列中删除一些（也可以不删除）元素来获得，剩下的元素保持其原始顺序。

• 输入：nums = [1,7,4,9,2,5]
• 输出：6
• 解释：整个序列均为摆动序列，各元素之间的差值为 (6, -3, 5, -7, 3)

乍一看，这道题求的不就是峰值个数嘛？

• 连续递增或者连续递减的不能够算摆动序列。并且数据是可以删除的。
• 那么我们把这种连续递增或者递减的数字删除掉，剩下的数字集合，画成一个折线图，就是高–>低–>高–>低–>高–>低这样的规律。

那么本题可以考虑用贪心算法：

• 局部最优：我们删除连续递增/递减上的节点（不算头和尾），那么此时就有两个峰值。一个峰顶一个峰谷。
• 全局最优：整个序列就有了最多的局部峰值，达到最长摆动序列。

那么我们一次遍历就可以完成上面的做法：

1. 从头开始遍历，记录当前节点和下一个节点的差值。假设为负数。
2. 那么下一次遍历的时候，如果差值为正数，摆动序列长度+1。否则跳过本次循环（连续递增/递减）

即判断当前节点满足下列条件的其中一种即可：

• 峰谷：当前值比上一个值小 preDiff <0 。下一个值比当前值大，curDiff >0。
• 峰顶：当前值比上一个值大 preDiff >0 。下一个值比当前值小，curDiff <0。

那么代码如下：

public int wiggleMaxLength(int[] nums) {
        if (nums.length < 2) {
            return nums.length;
        }
        int result = 1;
        int preDiff = 0;
        int curDiff = 0;
        for (int i = 0; i < nums.length - 1; i++) {
            // 下一个节点减去当前节点的值
            curDiff = nums[i + 1] - nums[i];
            // 如果curDiff >0 代表下一个节点比当前节点大。同时需要满足当前节点比上一个节点小。即峰谷
            if (curDiff > 0 && preDiff <= 0) {
                preDiff = curDiff;
                result++;
            }
            // 当前节点为峰顶
            if (curDiff < 0 && preDiff >= 0) {
                preDiff = curDiff;
                result++;
            }
        }
        return result;
    }

备注：为了更直观，if 写了两个，实际上可以把它们整合到一起。