leetcode-376.摆动序列
贪心算法
题目详情
如果连续数字之间的差严格地在正数和负数之间交替,则数字序列称为 摆动序列 。第一个差(如果存在的话)可能是正数或负数。仅有一个元素或者含两个不等元素的序列也视作摆动序列。
例如, [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
我的代码:
class Solution
{
public:
int wiggleMaxLength(vector<int>& nums)
{
int pre=0,cur=0,ans=1; //注意ans默认为1
if(nums.size()<=1) //若仅含一个元素
return nums.size();
for(int i=0;i<nums.size()-1;++i)
{
cur=nums[i+1]-nums[i];
if(pre<=0&&cur>0||pre>=0&&cur<0) //这里pre有等号的原因是
//满足两个不等元素情况,也可以单独处理
{
++ans;
pre=cur;
}
}
return ans;
}
};
涉及知识点:
1.贪心算法
顾名思义,贪心算法或贪心思想采用贪心的策略,保证每次操作都是局部最优的,从而使最
后得到的结果是全局最优的。
思路:
因为题目要求的是最长摆动子序列的长度,所以只需要统计数组的峰值数量就可以了
但是我们又要考虑,如果数组只有两个不等元素,那么result应该为2,所以我们每次讨论,
都将pre初始化为0,result初始化为1(默认最右边是一个峰值),默认在第一个数后面添加一个相同的数,然后在比较cur和pre时加上等号
就可以处理只有两个元素的情况
本题还可以利用动态规划的方法,利用二维dp
对于我们当前考虑的这个数,要么是作为山峰(即nums[i] > nums[i-1]
),要么是作为山谷(即nums[i] < nums[i - 1]
)。
设dp状态dp[i][0]
,表示考虑前i个数,第i个数作为山峰的摆动子序列的最长长度
设dp状态dp[i][1]
,表示考虑前i个数,第i个数作为山谷的摆动子序列的最长长度
dp[i][0] = max(dp[i][0], dp[j][1] + 1)
,其中0 < j < i且nums[j] < nums[i],表示将nums[i]
接到前面某个山谷后面,作为山峰。
dp[i][1] = max(dp[i][1], dp[j][0] + 1)
,其中0 < j < i且nums[j] > nums[i],表示将nums[i]
接到前面某个山峰后面,作为山谷。
初始状态:
由于一个数可以接到前面的某个数后面,也可以以自身为子序列的起点,所以初始状态为:dp[0][0] = dp[0][1] = 1
代码如下:
class Solution {
public:
int dp[1005][2];
int wiggleMaxLength(vector<int>& nums) {
memset(dp, 0, sizeof dp);
dp[0][0] = dp[0][1] = 1;
for (int i = 1; i < nums.size(); ++i)
{
dp[i][0] = dp[i][1] = 1;
for (int j = 0; j < i; ++j)
{
if (nums[j] > nums[i]) dp[i][1] = max(dp[i][1], dp[j][0] + 1);
}
for (int j = 0; j < i; ++j)
{
if (nums[j] < nums[i]) dp[i][0] = max(dp[i][0], dp[j][1] + 1);
}
}
return max(dp[nums.size() - 1][0], dp[nums.size() - 1][1]);
}
};