题目
官方思路:
我们需要找到最长摆动序列。一个摆动序列是一个上升下降交替的子序列。
为了更好地理解这一方法,用两个数组来 dp ,分别记作 up 和 down 。
每当我们选择一个元素作为摆动序列的一部分时,这个元素要么是上升的,要么是下降的,这取决于前一个元素的大小。
up[i]存的是目前为止最长的以第 i 个元素结尾(不包含i这个元素)的上升摆动序列的长度。
类似的, down[i]记录的是目前为止最长的以第 i 个元素结尾(不包含i这个元素)的下降摆动序列的长度。
我们每当找到将第 i 个元素作为上升摆动序列的尾部的时候就更新 up[i] 。现在我们考虑如何更新 up[i],我们需要考虑前面所有的降序结尾摆动序列,也就是找到down[j] ,满足 j < i 且 nums[i] >nums[j]。类似的, down[i]也会被更新。
代码
public class Solution {
public int wiggleMaxLength(int[] nums) {
if (nums.length < 2)
return nums.length;
int[] up = new int[nums.length];
int[] down = new int[nums.length];
for (int i = 1; i < nums.length; i++) {
for(int j = 0; j < i; j++) {
if (nums[i] > nums[j]) {
up[i] = Math.max(up[i],down[j] + 1);
} else if (nums[i] < nums[j]) {
down[i] = Math.max(down[i],up[j] + 1);
}
// 上面更新up和down时要+1的原因是:
//前面已提到,up[i]和down[j]存的是以i结尾的不包含i的最大长度,所以+1是加上j这个元素,所以要在[j]+1
}
}
//+1是因为循环 j < i j会不等于i
return 1 + Math.max(down[nums.length - 1], up[nums.length - 1]);
}
}