leetcode376. 摆动序列

cv工程师^ ^

已于 2023-10-15 20:03:18 修改

阅读量71

点赞数

分类专栏：算法题解文章标签：算法

于 2023-10-15 19:53:32 首次发布

本文链接：https://blog.csdn.net/rangbiubiu/article/details/133846588

版权

算法题解专栏收录该内容

11 篇文章 0 订阅

订阅专栏

关于删除元素？

因为题目并未要求我们求出最长摆动子序列，而是只要求返回其长度，因此我们不需要删除导致子序列不为摆动序列的元素，而是直接在遍历序列的时候跳过它们即可。

求最长摆动子序列的长度

若将摆动序列化成折线图，则摆动序列的折线图是一上一下一上一下反复交替的，即折线图仅由峰/谷构成（一个摆动是一个峰/谷），没有平坡。

‘峰’ 指山峰，'谷 ’ 指谷底

请添加图片描述

设最长摆动序列有x个峰/谷，则算上首尾元素，摆动序列的长度= x+2。
但是若nums数组的峰/谷个数=0呢？无峰/谷则nums数组的折线图有两种情况：

折线图为一条水平直线，即数组所有元素相等。则该水平直线可以抽象为一个点，其最长摆动序列的长度同只有一个元素时一样，都等于1
折线图为一条斜直线，即数组元素单调递增或单调递减。则该斜直线可以抽象为由两个端点构成的线段，其最长摆动子序列的长度同只有两个不等元素时一样，都等于2

请添加图片描述

总结

计算最长摆动子序列的长度：
用swingCnt记录整个数组的摆动次数，每一次摆动swingCnt就++。
如果遍历完整个数组后swingCnt仍=0，说明数组没有一个摆动，即无峰/谷，按上述所说的无峰/谷的两种情况计算即可；若swingCnt != 0则摆动序列的长度=swingCnt+2

遍历nums数组

现在我们可以知道，nums数组的折线图由平坡和峰/谷组成，要想求摆动序列的长度就要求整个数组的摆动次数swingCnt，也即峰/谷个数。

我们用prediff 和 lastdiff分别表示【当前遍历元素与前一个元素的差】、【后一个元素与当前遍历元素的差】。

当prediff < 0 && lastdiff > 0 || prediff > 0 && lastdiff < 0时当前遍历元素和前后元素之间就形成了一个峰/谷，算一次摆动，swingCnt++。

除此之外就是有平坡的情况，而平坡有两种情况：

先上坡再平坡再下坡（先下坡再平坡再上坡也一样），将这种情况记为a
先上坡再平坡再上坡（先下坡再平坡再下坡也一样），将这种情况记为b

请添加图片描述

🧐为什么要分这两种情况处理平坡呢？
因为这两种情况计算摆动不一样，对于情况a，10-200-200-100算一次摆动，数组的摆动序列的长度=3+2=5；而对于情况b，10-200-200-220不算摆动，数组的摆动序列的长度=2+2=4；

接下来我们分析一下这两种情况应该怎么写代码。

对于情况a，当当前遍历元素为第一个200时，prediff>0，lastdiff=0，继续走遍历到第二个200时，prediff=0，lastdiff<0。也就是说是 > = <，即上坡平坡下坡，这种情况算一次摆动
对于情况b，当当前遍历元素为第一个200时，prediff>0，lastdiff=0，继续走遍历到第二个200时，prediff=0，lastdiff>0。也就是说是 > = >，即上坡平坡上坡，这种情况不算摆动

是因此我们发现，若出现平坡时，我们不将prediff更新为0，而是保持原来的值不变，那么就能让swingCnt正确++，正确计算摆动次数。

即当遇到平坡时，lastdiff始终不断往右推进但是prediff始终在平坡前的陡坡上，直到遍历到平坡后的下一个陡坡

如情况a，就从 > = < 变成了 > <，即prediff < 0 && lastdiff > 0，swingCnt++
如情况b，就从 > = >编程了 > >，即prediff > 0 && lastdiff > 0，swingCnt不做++

🎈总结：什么时候更新prediff ? 只要出现了峰/谷就更新prediff为lastdiff，否则不做更新！

这样我们就把峰/谷和平坡的处理情况都搞清楚了。接下来上代码！
呃不是，还要注意一下，prediff的初始值！
比如nums={3,3,3,2,5}，若不做处理则prediff初始值=0，但是我们只有在出现峰/谷(prediff < 0 && lastdiff > 0)时才会更新prediff为lastdiff，因此随着遍历，lastdiff不断推进，但是prediff始终更新不了，始终为0。
因此我们要将prediff初始化为第一个不为0的值再开始遍历计算峰/谷数，即从第一个陡坡开始遍历。

案例case

这些案例包含了所有情况（看完上面可以结合案例先自己再做一次）：

{1,17,5,10,13,15,10,5,16,8} 输出7
{40,240,10,200,200,100} 输出5
{40,240,10,200,200,220} 输出4
{1,2,3,4,5} 输出2
{3,3,3,2,5} 输出3
{1,1,1} 输出1

代码

    int wiggleMaxLength(vector<int>& nums) {
        int size = nums.size(), swingCnt = 0;//摆动数（峰/谷数）

        if(size == 1) return 1;

        int j = 0;
        while (j < size - 1 && nums[j] == nums[j+1]) ++j;
        //不能写成 nums[j] == nums[j+1] && j < size - 1！

        if (j == size - 1) return 1;//说明数组所有元素都相等

        int prediff = nums[j + 1] - nums[j];//初始prediff不能为0，要从陡坡开始

        for (int i = 1; i < nums.size() - 1; ++i) {
            int lastdiff = nums[i + 1] - nums[i];
            if(prediff < 0 && lastdiff > 0 || prediff > 0 && lastdiff < 0) {
                ++swingCnt;
                prediff = lastdiff;
            }
        }
        int Len_swingSq = swingCnt != 0 ? swingCnt + 2 : 2;//摆动子序列的长度
        return Len_swingSq;
    }

细节：

while (j < size - 1 && nums[j] == nums[j+1])不能写成nums[j] == nums[j+1] && j < size - 1，否则每次判断会先执行第一个式子nums[j] == nums[j+1]，当j=size-1时j+1=size就已经越界了，直接就报错了，而不会去判断第二个式子 j < size - 1

如果对你有帮助的话就太好啦ヾ(≧▽≦*)o