376. 摆动序列https://leetcode.cn/problems/wiggle-subsequence/description/
如果连续数字之间的差严格地在正数和负数之间交替,则数字序列称为摆动序列。第一个差(如果存在的话)可能是正数或负数。仅有一个元素或者含两个不等元素的序列也视作摆动序列。例如,[1, 7, 4, 9, 2, 5]是一个摆动序列,因为差值(6, -3, 5, -7, 3)是正负交替出现的。相反,[1, 4, 7, 2, 5]和[1, 7, 4, 5, 5]不是摆动序列,第一个序列是因为它的前两个差值都是正数,第二个序列是因为它的最后一个差值为零。子序列可以通过从原始序列中删除一些(也可以不删除)元素来获得,剩下的元素保持其原始顺序。给你一个整数数组nums,返回nums中作为摆动序列的最长子序列的长度。
- 输入:nums = [1,7,4,9,2,5],输出:6,解释:整个序列均为摆动序列,各元素之间的差值为(6, -3, 5, -7, 3)。
- 输入: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)。
- 输入:nums = [1,2,3,4,5,6,7,8,9],输出:2。
提示:1 <= nums.length <= 1000,0 <= nums[i] <= 1000。
进阶:你能否用O(n)的时间复杂度完成此题?
我们用动态规划的思想来解决这个问题。
确定状态表示:根据经验和题目要求,为了方便推导状态转移方程,我们定义如下的状态表示:
- 用f[i]表示:所有以i位置为结尾的最后呈现上升趋势的子序列中,最长的摆动序列的长度。
- 用g[i]表示:所有以i位置为结尾的最后呈现下降趋势的子序列中,最长的摆动序列的长度。
推导状态转移方程:考虑f[i]。我们把以i位置为结尾的子序列分为2类:长度为1的子序列,长度大于1的子序列。如果子序列的长度是1,那么这个子序列是摆动序列。下面我们考虑长度大于1的子序列。
如果以i位置为结尾的子序列的长度大于1,我们可以继续细分为:i位置元素的前面是i - 1位置元素的子序列,i位置元素的前面是i - 2位置元素的子序列,i位置元素的前面是i - 3位置元素的子序列,……,i位置元素的前面是0位置元素的子序列。也就是说,如果子序列中,i位置元素的前面是j位置元素,那么j的范围是[0, i - 1]。
对于每一个j,如果nums[j] < nums[i],那么这个序列最后呈现上升趋势。若想要这个序列是摆动序列,就要找到以j位置为结尾的最后呈现下降趋势的摆动序列,在后面添加上nums[i],就是以i位置为结尾的最后呈现上升趋势的摆动序列。若想要这个摆动序列尽可能得长,那么以j位置为结尾的最后呈现下降趋势的摆动序列就要尽可能得长,根据状态表示,以j位置为结尾的最后呈现下降趋势的摆动序列的最长长度是g[j]。所以,以i位置为结尾的最后呈现上升趋势的摆动序列的最长长度是g[j] + 1。
我们来梳理一下。f[i]应该取:「1」和「所有的满足0 <= j < i并且nums[j] < nums[i]的j中,g[j] + 1的最大值」的较大值。为了实现这个逻辑,我们可以把f表的值全部初始化为1,接着让i > 0时的f[i]取:所有的满足0 <= j < i并且nums[j] < nums[i]的j中,g[j] + 1的最大值。
同理,g[i]应该取:「1」和「所有的满足0 <= j < i并且nums[j] > nums[i]的j中,f[j] + 1的最大值」的较大值。为了实现这个逻辑,我们可以把g表的值全部初始化为1,接着让i > 0时的g[i]取:所有的满足0 <= j < i并且nums[j] > nums[i]的j中,f[j] + 1的最大值。
填表顺序:根据状态转移方程,显然应从左往右同时填2个表。
返回值:根据状态表示,我们应返回f表和g表的所有值中的最大值。
细节问题:f表和g表的规模和nums相同,均为1 x n。
class Solution {
public:
int wiggleMaxLength(vector<int>& nums) {
int n = nums.size();
// 创建dp表
vector<int> f(n, 1);
auto g = f;
// 填表
for (int i = 1; i < n; i++) {
for (int j = 0; j < i; j++) {
if (nums[j] < nums[i]) {
f[i] = max(f[i], g[j] + 1);
} else if (nums[j] > nums[i]) {
g[i] = max(g[i], f[j] + 1);
}
}
}
return max(*max_element(f.begin(), f.end()),
*max_element(g.begin(), g.end()));
}
};