LeetCode 376. Wiggle Subsequence

问题描述

• 地址

问题分析

• 在 O(N) 的时间复杂度下求一个数组的最长摆动（升序降序间或变化）子序列长度。
• 用贪心的思想，用up 来记录上一个子序列末尾两位数字是升序的最长子序列的长度，用down 来记录上一个子序列末尾两位数字是降序的最长子序列的长度，初始化都为1，从i = 1 开始向后遍历nums数组:
  
  • 如果当前nums[i] 是升序，那么便更新up,up = down + 1
  • 如果当前nums[i] 是降序，那么便更新down,down = up + 1
    最终，该数组的最长摆动子序列的长度便是 Math.max(up,down)

经验教训

• 理解这种贪心做法为什么对

代码实现

class Solution {
    public int wiggleMaxLength(int[] nums) {
        if (nums == null) {
            return 0;
        }
        if (nums.length <= 1) {
            return nums.length;
        }
        int up = 1;
        int down = 1;
        for(int i = 1; i < nums.length; ++i) {
            int diff = nums[i] - nums[i - 1];
            if (diff > 0) {
                up = down + 1;
            }
            if (diff < 0) {
                down = up + 1;
            }
        }
        return Math.max(up, down);
    }
}