力扣刷题笔记：376. 摆动序列

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

例如， [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
解释: 整个序列均为摆动序列。

思路：来自力扣题解的，我在其讲解的基础上加入自己的理解

• 这一道题可以用贪心算法，以O(n) 的时间复杂度 和 O(1) 的空间复杂度写出

• count：计数； prevdiff：维持前面的状态；diff：维持当前的状态
  

• 首先我们把每个元素的值和下标画出这样的图，我们可以发现，不管在上升和下降的过程中，只要在连续上升（下降）的过程中元素个数超过两个的话，我们只取两个即可

• 所以我们遍历中只需要判断前面的状态和当前的状态即可

• 这样子，在连续上升（下降）的过程，我们根据上面的步骤，我们就加了两次

• 虽然没有按题目要求对应的元素给count++，但这也是一种看透题目本质的算法

源代码:

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        int n = nums.size();

        if(n < 2){
            return n;
        }
        int prevdiff = nums[1] - nums[0];
        int count = prevdiff != 0 ? 2 : 1;
        for (int i = 2; i < n; i++) {
            int diff = nums[i] - nums[i - 1];
            if ((diff > 0 && prevdiff <= 0) || (diff < 0 && prevdiff >= 0)) {
                count++;
                prevdiff = diff;
            }
        }
        return count;
    }
};