leetcode: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
解释: 整个序列均为摆动序列。
示例 2:

输入: [1,17,5,10,13,15,10,5,16,8]
输出: 7
解释: 这个序列包含几个长度为 7 摆动序列,其中一个可为[1,17,10,13,10,16,8]。
示例 3:

输入: [1,2,3,4,5,6,7,8,9]
输出: 2
进阶:
你能否用 O(n) 时间复杂度完成此题?

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/wiggle-subsequence
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路:
最优解法是使用贪心,简单的理解,将序列看作折线图,因此我们要求的则是折线拐点数量加一。步骤如下:

  1. 开始循环,遍历数组的每一个节点,使用count来记录拐点的数量;
  2. for循环体内部使用while循环将连续相等的节点排除;
  3. 使用flag来记住在此之前的序列是递增还是递减的,以此判断当前是否是拐点;
  4. 若是拐点则将count++;
  5. 需要注意的点:
    1. 当数组为全相等的情况,使用flag置0来排除影响;
    2. 当数组的开头有相等的情况,内层的while循环来排除;

新的理解:
在写代码的时候脑子里面要有清晰的思路,没有思路的时候尽量想清楚之后再写,比如说这道题,需要想清楚的有,如何处理边界值,处理数组元素全都相等的情况,处理上升和下降序列的判断,处理开头元素相等带来的count值的错误。

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        int len = nums.size();
        int count = 1;
        if(len==0)return 0;
        if(len==1)return 1;
        int flag = 0;
        int temp = 0;
        for(int i = 0;i<len-1;i++)
        {
            while(i<len-1&&nums[i]==nums[i+1])
            {
                i++;
                if(i==len-1)return count;
            }
            int sub = nums[i+1]-nums[i];
            if(flag==0)
            {
                flag=sub>0?1:-1;
                count++;
                continue;
            }else if(sub*flag<0)
            {
                count++;
                flag = -flag;
            }
            
        }
        return count;
    }
};

你可能感兴趣的:(leetcode)