题目描述:
如果连续数字之间的差严格地在正数和负数之间交替,则数字序列称为摆动序列。第一个差(如果存在的话)可能是正数或负数。少于两个元素的序列也是摆动序列。例如, [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
解释: 整个序列均为摆动序列。
示例 2:
输入: [1,17,5,10,13,15,10,5,16,8]
输出: 7
解释: 这个序列包含几个长度为 7 摆动序列,其中一个可为[1,17,10,13,10,16,8]。
示例 3:
输入: [1,2,3,4,5,6,7,8,9]
输出: 2
进阶:你能否用 O(n) 时间复杂度完成此题?
思路一(dp):–时间O(n),空间O(n)
看到"最大子序列"这个关键词,第一个想到dp,但是怎么dp是个难点。
既是摆动序列,序列会上下起伏,
- up[i]记录上升沿时,以i结尾最大摆动序列长度;
- 类似的,down[i]记录下降沿时,以i结尾最大摆动序列长度;
- 最后返回max(up[n-1],down[n-1])
class Solution {
public:
int wiggleMaxLength(vector<int>& nums) {
if(nums.size()<2) return nums.size();
int n=nums.size();
vector<int> up(n,0);//up[i]上升沿时,以i结尾最大摆动序列长度
vector<int> down(n,0);//down[i]下降沿时,以i结尾最大摆动序列长度
up[0]=1;
down[0]=1;
for(int i=1;i<n;++i){
if(nums[i]>nums[i-1]){//up
up[i]=down[i-1]+1;
down[i]=down[i-1];
}
else if(nums[i]<nums[i-1]){//down
down[i]=up[i-1]+1;
up[i]=up[i-1];
}
else if(nums[i]==nums[i-1]){//keep
up[i]=up[i-1];
down[i]=down[i-1];
}
}
return max(up[n-1],down[n-1]);
}
};
空间优化 --O(1)
class Solution {
public:
int wiggleMaxLength(vector<int>& nums) {
if(nums.size()<2) return nums.size();
int n=nums.size();
int up=1,down=1;
for(int i=1;i<n;++i){
if(nums[i]>nums[i-1]){//up
up=down+1;
}
else if(nums[i]<nums[i-1]){//down
down=up+1;
}
}
return max(up,down);
}
};
思路二:(贪心)–时间O(n),空间O(1)
看一个例子,[1,17,5,10,13,15,10,5,16,8],当数组遍历到第四位时,即
[1,17,5,10…时,5,10,13,15形成上升序列,显然这四个数只需选一个作为摇摆序列的一部分。
选择哪个呢?若选择10,则15后面的10与其不会摇摆,故应该选大的,这样后面摇摆的概率更高!
这是贪心的思想。
再看一个图片 source of picture
从上图中也可以看出,如果选择C作为摆动序列的第二个点,则无法包括E进去。
因此本题等价于寻找数组中交替的最大最小值。
class Solution {
public:
int wiggleMaxLength(vector<int>& nums) {
if(nums.size()<2) return nums.size();
int pre_diff=nums[1]-nums[0];
int cnt;
pre_diff==0?cnt=1:cnt=2;
for(int i=2;i<nums.size();++i){
int diff=nums[i]-nums[i-1];
//上升序列碰到下降沿 或 下降序列碰到上升沿
if(pre_diff>=0 && diff<0 || pre_diff<=0 && diff>0){
cnt++;
pre_diff=diff;//只有摇摆时才更新pre_diff
//如果时时更新遇到diff=0会丢失序列状态
}
}
return cnt;
}
};