376. 摆动序列（贪心）

题目

376. 摆动序列

思路

1. 首先单独处理前两个元素，因为如果它们相等的话只能算一个，不相等的话算是两个，这是计数的基础
2. 处理后续元素，这里因为有了preDiff = 0的情况，所以不能单纯的用preDiff * curDiff <=0 来判断，因为后续我们一旦找到curDiff = 0的时候，是不能计数的，但是用了乘法，就会多计数进去；
3. 如果进入循环的条件不包含preDiff = 0， 那么前两个元素相同的数组答案会全部为1。
4. 除了第一段差，不可能再有preDiff = 0的情况，因为条件里控制了curDiff > 0 或者curDiff < 0

代码

/**
 * @param {number[]} nums
 * @return {number}
 */
var wiggleMaxLength = function(nums) {
    if(nums.length === 1) return 1;
    //处理前两个元素
    let preDiff = nums[1] - nums[0];
    let count = preDiff === 0 ? 1:2;
    //注意这个地方不需要把长度正好是2的单独拿出来讨论，因为长度是2的不会进入循环
    for(let i = 2; i < nums.length; i++){
        let curDiff = nums[i] - nums[i - 1];
        if(curDiff > 0 && preDiff <= 0 || curDiff < 0 && preDiff >= 0){
            count ++;
            preDiff = curDiff;
        }
    }
    return count;
};