1. 解析
题目大意,求最长的摆动子序列长度,可以认为 Wiggle Sort II 的进阶版
2. 分析
一般求解这种子序列的题目,都是DP的题。摆动序列的状态一共分为两种,第一种:正, 负, 正, 负, .......;第二种:负, 正, 负, 正, ......;所以整体上和Longest Increasing Subsequence不同点在于,当前状态不仅仅只有一种,所以我们要借助两个DP数组,positive[i]表示第一种摆动序列在范围[0, i]序列内的最大长度,negative[i]表示第二种摆动序列在范围[0, i]序列内的最大长度,状态转移方程为:
class Solution {
public:
int wiggleMaxLength(vector<int>& nums){
if (nums.size() <= 1) return nums.size();
vector<int> positive(nums.size(), 1);
vector<int> negative = positive;
for (int i = 1; i < nums.size(); ++i){
for (int j = 0; j < i; j++){
if (nums[j] < nums[i]) positive[i] = max(positive[i], negative[j] + 1);
else if (nums[j] > nums[i]) negative[i] = max(negative[i], positive[j] + 1);
}
}
return max(positive.back(), negative.back());
}
};
还有另外一种思路, 利用pair键值对存储当前摆动序列的最大长度和对应的当前状态(0------当前为待匹配元素,1------当前为递增,-1-------当前为递减。dp[i]表示前i个序列的最大长度摆动序列和对应的状态;当且仅当两个状态不一样(摆动)而且当前长度小于待匹配长度加1时,进行更新。
class Solution {
public:
int wiggleMaxLength(vector<int>& nums) {
if (nums.size() <= 1) return nums.size();
int n = nums.size();
int res = 1;
vector<pair<int , int>> dp(n, {1, 0});
for (int i = 1; i < n; ++i){
for (int j = i - 1; j >= 0; --j){
if (nums[i] == nums[j]) continue;
int flag = nums[i] - nums[j] > 0 ? 1 : -1; //标记当前的状态
if (dp[j].second != flag && dp[j].first + 1 > dp[i].first){ //更新当前最优值
dp[i].first = dp[j].first + 1;
dp[i].second = flag;
}
}
res = max(res, dp[i].first);
}
return res;
}
};