题目
如果连续数字之间的差严格地在正数和负数之间交替,则数字序列称为摆动序列。第一个差(如果存在的话)可能是正数或负数。少于两个元素的序列也是摆动序列。
例如, [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
解释: 整个序列均为摆动序列。
想法
动态规划
这道题类似于求最长上升子序列。因为涉及到差的正负数,所以设置两个变量,设up代表正值(即第二个数>第一个数,nums[i]>nums[i-1]),down代表负数(即第二个数<第一个数,nums[i]<nums[i-1]).up和down初始值设置为1,up=down=1(只有一个数的情况)
举例: [1,7,4,9,2,5]
i=0,nums[i]=1,up=down=1
i=1,nums[i]=7,nums[i]>nums[i-1]说明这是一个上升序列,up=down+1(上升序列的前面肯定是下降序列,因为题目要求就是正负交替)
i=2,nums[i]=4,nums[i]<nums[i-1]说明这是一个下降序列,down=up+1(理由同上)
......
最后可以得出结果。
class Solution {
public int wiggleMaxLength(int[] nums) {
//摆动序列:两数之差在正数负数之间交替
if(nums.length < 2) return nums.length;
int up = 1, down = 1;
for(int i = 1; i < nums.length; i++){
if(nums[i] > nums[i-1]){
up = down + 1;
}else if(nums[i] < nums[i-1]){
down = up + 1;
}
}
return Math.max(up,down);
}
}