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<Integer> st = new Stack<Integer>();
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