Leetcode - Wiggle Subsequence

My code:

public class Solution {
    public int wiggleMaxLength(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        else if (nums.length == 1) {
            return 1;
        }
        
        boolean nextBig = false;
        for (int i = 1; i < nums.length; i++) {
            if (nums[i] > nums[0]) {
                nextBig = true;
                break;
            }
            else if (nums[i] < nums[0]) {
                nextBig = false;
                break;
            }
        }
        
        Stack st = new Stack();
        st.push(nums[0]);
        for (int i = 1; i < nums.length; i++) {
            if (nums[i] > st.peek()) {
                if (nextBig) {
                    st.push(nums[i]);
                    nextBig = !nextBig;
                }
                else {
                    st.pop();
                    st.push(nums[i]);
                }
            }
            else if (nums[i] < st.peek()) {
                if (nextBig) {
                    st.pop();
                    st.push(nums[i]);
                }
                else {
                    st.push(nums[i]);
                    nextBig = !nextBig;
                }
            }
        }
        
        return st.size();
    }
}

reference:
https://discuss.leetcode.com/topic/51946/very-simple-java-solution-with-detail-explanation/2

这道题目没能自己做出来。
哎,发现新题,稍微难一些,基本就做不出来了。

这道题目官方给的题解,用了很多方法:
https://leetcode.com/articles/wiggle-subsequence/

DP那些实在有些看不动了。大脑有点累,转不动了。

Greedy 的思想最容易理解。
就是先确定 index 0,1 的大小关系。
比如 num1[0] > nums[1] , 那么下面我们就需要 nums[1] < nums[2]
如果 nums[2] < nums[1] ,那就把 nums[1] delete, 加入 nums[2]

就这样,保证顶部的永远是 max / min peak

这道题目可以不用栈,用一个 pre variable 就能存。
但如果有 follow up 要求还原出这个 array,那就必须得用栈来做记录了。

Anyway, Good luck, Richardo! -- 09/21/2016

你可能感兴趣的:(Leetcode - Wiggle Subsequence)