原题地址:https://leetcode-cn.com/problems/wiggle-subsequence/submissions/
题目描述:
给定一个整数序列,返回作为摆动序列的最长子序列的长度。 通过从原始序列中删除一些(也可以不删除)元素来获得子序列,剩下的元素保持其原始顺序。
示例 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) 时间复杂度完成此题?
解题方案:
使用动态规划和贪心算法解题,解题的思路不太难想,采用两个数组作为动态规划数组。
代码:
class Solution {
public:
int wiggleMaxLength(vector<int>& nums) {
if(nums.size() == 0 || nums.size() == 1)
return nums.size();
vector<int> up(nums.size(), 1);
vector<int> down(nums.size(), 1);
for(int i = 1; i < nums.size(); i ++){
if(nums[i] > nums[i - 1]){
up[i] = down[i - 1] + 1;
down[i] = down[i - 1];
}else if(nums[i] < nums[i - 1]){
down[i] = up[i - 1] + 1;
up[i] = up[i - 1];
}else{
down[i] = down[i - 1];
up[i] = up[i - 1];
}
}
return max(down[nums.size() - 1], up[nums.size() - 1]);
}
};
给出时间最短的算法:
class Solution {
public:
int wiggleMaxLength(vector<int>& nums) {
if(nums.size()<2)
return nums.size();
static const int BEGIN=0;
static const int UP=1;
static const int DOWN=2;
int STATE=BEGIN;
int max_length=1;
for(int i=1;i<nums.size();++i)
{
switch(STATE){
case BEGIN:
if(nums[i]>nums[i-1])
{
STATE=UP;
max_length++;
}
else if(nums[i]<nums[i-1])
{
STATE=DOWN;
max_length++;
}
break;
case UP:
if(nums[i-1]>nums[i])
{
STATE=DOWN;
max_length++;
}
break;
case DOWN:
if(nums[i-1]<nums[i])
{
STATE=UP;
max_length++;
}
break;
}
}
return max_length;
}
};