持续创作,加速成长!这是我参与「掘金日新计划 · 10 月更文挑战」的第24天,点击查看活动详情
最近一直在力扣刷题,也逐渐对各类题型有了自己的理解,所谓见招拆招,将自己的浅显经验分享一下,帮助更多在编程路上的朋友们。
摆动序列
如果连续数字之间的差严格地在正数和负数之间交替,则数字序列称为 摆动序列 。 第一个差(如果存在的话)可能是正数或负数。仅有一个元素或者含两个不等元素的序列也视作摆动序列。
- 例如,
[1, 7, 4, 9, 2, 5]是一个 摆动序列 ,因为差值(6, -3, 5, -7, 3)是正负交替出现的。 - 相反,
[1, 4, 7, 2, 5]和[1, 7, 4, 5, 5]不是摆动序列,第一个序列是因为它的前两个差值都是正数,第二个序列是因为它的最后一个差值为零。
子序列 可以通过从原始序列中删除一些(也可以不删除)元素来获得,剩下的元素保持其原始顺序。
给你一个整数数组 nums ,返回 nums 中作为 摆动序列 的 最长子序列的长度 。
示例 1:
输入: nums = [1,7,4,9,2,5]
输出: 6
解释: 整个序列均为摆动序列,各元素之间的差值为 (6, -3, 5, -7, 3) 。
示例 2:
输入: nums = [1,17,5,10,13,15,10,5,16,8]
输出: 7
解释: 这个序列包含几个长度为 7 摆动序列。
其中一个是 [1, 17, 10, 13, 10, 16, 8] ,各元素之间的差值为 (16, -7, 3, -3, 6, -8) 。
示例 3:
输入: nums = [1,2,3,4,5,6,7,8,9]
输出: 2
思路
该题的难点主要在于dp的定义上,需要两个数组分别保存数组差的状态,定义up为最后状态为递增的摆动序列,down为最后最后为递减的摆动序列。
初始化一个元素也默认为摆动序列,如果差值正负相同,则最大长度不变,如果差值正负不同,则取相反的状态的长度+1。
最终结果可能是以递增结尾,也可能是以递减结尾,取二者中长度最大的即为答案。
题解
class Solution {
public int wiggleMaxLength(int[] nums) {
int n = nums.length;
int[] up = new int[n];
int[] down = new int[n];
up[0] = 1;
down[0] = 1;
for(int i = 1; i < n; i++) {
if(nums[i] - nums[i - 1] > 0) {
up[i] = Math.max(up[i - 1], down[i - 1] + 1);
down[i] = down[i - 1];
}else if(nums[i] - nums[i - 1] < 0) {
up[i] = up[i - 1];
down[i] = Math.max(down[i -1], up[i - 1] + 1);
}else {
up[i] = up[i - 1];
down[i] = down[i - 1];
}
}
return Math.max(up[n - 1], down[n - 1]);
}
}