每日算法系列【LeetCode 376】摆动序列

关注公众号【算法码上来】,每日算法干货马上就来!

题目描述

如果连续数字之间的差严格地在正数和负数之间交替,则数字序列称为摆动序列。第一个差(如果存在的话)可能是正数或负数。少于两个元素的序列也是摆动序列。

例如, [1,7,4,9,2,5] 是一个摆动序列,因为差值 [6,-3,5,-7,3] 是正负交替出现的。相反, [1,4,7,2,5] 和 [1,7,4,5,5] 不是摆动序列,第一个序列是因为它的前两个差值都是正数,第二个序列是因为它的最后一个差值为零。

给定一个整数序列,返回作为摆动序列的最长子序列的长度。 通过从原始序列中删除一些(也可以不删除)元素来获得子序列,剩下的元素保持其原始顺序。

示例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

题解

这题题面说的啰里啰唆的,其实就一句话:给你一个序列,找出最长的一个子序列,其中子序列相邻两个数的大小是波形的(也就是大小大小大等等这样的)。

暴力法

用 dfs 枚举所有可能的子序列,然后看最长的是多少,这种方法显然会超时。

动态规划

其实看到这道题,我第一个想到了最长上升子序列,这不就变了个形式嘛,于是动态规划解法直接就有了。

用 $dp[i][s]$ 表示以 $nums[i]$ 结尾的符合条件的最长子序列长度,其中 s 取 1 表示在 $nums[i]$ 处子序列上升,取 0 表示下降。那么我们只需要遍历之前的所有 j ,如果 $nums[j] < nums[i]$ ,那么在 j 处必须是要下降的,更新:
$$
dp[i][1] = \max_j{\{dp[j][0]\}}
$$
如果 $nums[j] > nums[i]$ ,那么在 j 处必须是要上升的,更新:
$$
dp[i][0] = \max_j{\{dp[j][1]\}}
$$
然后取数组中最大值就是答案了,时间复杂度 $O(n^2)$ 。

动态规划+时间优化

换个定义,用 $dp[i][s]$ 表示 $nums[i]$ 之前的最长子序列,注意和上面的区别就是不一定以 $nums[i]$ 结尾了。 s 取 1 表示最后两个数是上升的,取 0 表示最后两个数是下降的。

这里分为几种情况:

  • $nums[i-1] < nums[i]$ :
    • 考虑 $dp[i][0]$ ,也就是最后两个数下降的,那肯定不能取 $nums[i]$ ,因为 $nums[i-1]$ 比它更小、更优,所以直接更新为 $dp[i-1][0]$ 。
    • 考虑 $dp[i][1]$ ,也就是最后两个数上升的,那如果不取 $nums[i]$ ,那更新为 $dp[i-1][1]$ ;如果取的话,我们就要保证 i-1 之前最后两个数是下降的,并且之前的最后一个数小于 $nums[i]$ 。我们可以证明, i-1 之前的最后两个下降的数一定满足:第二个数 $nums[j]$ 是小于 $nums[i]$ 的,因为如果 $nums[j] >= nums[i]$ ,那么 j 到 i 之间的数一定是单调下降的,否则存在更长的子序列,那么就和 $nums[i-1] < nums[i]$ 矛盾了。综上,取的话 $dp[i][1]$ 更新为 $dp[i-1][0]+1$ 。
  • $nums[i-1] > nums[i]$ :
    同样考虑最后两个数上升还是下降,分析和上面一样。

综上考虑,时间复杂度可以降为 $O(n)$ ,空间复杂度是 $O(n)$ 。

动态规划+空间优化

在上面优化的基础上,我们还可以观察到,每一次 $dp[i][s]$ 其实只会用到 $dp[i-1][s]$ ,所以我们只需要保存当前和前一时刻的状态就行了,空间复杂度可以降为 $O(1)$ 。

贪心法

其实这题还可以直接贪心做,考虑一段连续的上升序列,最优子序列一定是包括了首尾两个数的,因为首是最小的数,选了它才能给前一个数留出更大的上升空间,而尾是最大的数,选了它才能给下一个数留出更多的下降空间。

所以我们贪心的扫描一遍数组,遇到上升或者下降的转折点就选取这个数。而如果数组不升不降,也就是不变的话,就不用管它,因为这些相同的数里面只需要选取一个就行了。

时间复杂度是 $O(n)$ ,空间复杂度是 $O(1)$ 。

代码

动态规划(c++)

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        int n = nums.size();
        if (n <= 1) return n;
        int dp[n][2], res = 1;
        memset(dp, 0, sizeof dp);
        dp[0][0] = dp[0][1] = 1;
        for (int i = 1; i < n; ++i) {
            for (int j = 0; j < i; ++j) {
                if (nums[j] != nums[i]) {
                    int s = nums[j] < nums[i];
                    dp[i][s] = max(dp[i][s], dp[j][s^1]+1);
                }
            }
            res = max(res, dp[i][0]);
            res = max(res, dp[i][1]);
        }
        return res;
    }
};

动态规划+时间优化(c++)

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        int n = nums.size();
        if (n <= 1) return n;
        int dp[n][2];
        memset(dp, 0, sizeof dp);
        dp[0][0] = dp[0][1] = 1;
        for (int i = 1; i < n; ++i) {
            if (nums[i] == nums[i-1]) {
                dp[i][0] = dp[i-1][0];
                dp[i][1] = dp[i-1][1];
            } else {
                int s = nums[i] > nums[i-1];
                dp[i][s] = max(dp[i-1][s], dp[i-1][s^1] + 1);
                dp[i][s^1] = dp[i-1][s^1];
            }
        }
        return max(dp[n-1][0], dp[n-1][1]);
    }
};

动态规划+空间优化(c++)

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        int n = nums.size();
        if (n <= 1) return n;
        int dp[2][2];
        memset(dp, 0, sizeof dp);
        dp[0][0] = dp[0][1] = 1;
        for (int i = 1; i < n; ++i) {
            if (nums[i] == nums[i-1]) {
                dp[1][0] = dp[0][0];
                dp[1][1] = dp[0][1];
            } else {
                int s = nums[i] > nums[i-1];
                dp[1][s] = max(dp[0][s], dp[0][s^1]+1);
                dp[1][s^1] = dp[0][s^1];
                swap(dp[0][s], dp[1][s]);
                swap(dp[0][s^1], dp[1][s^1]);
            }
        }
        return max(dp[0][0], dp[0][1]);
    }
};

贪心(c++)

class Solution {
public:
    int wiggleMaxLength(vector<int>& nums) {
        int n = nums.size();
        if (n <= 1) return n;
        int res = 1, pre_ord = -1;
        for (int i = 1; i < n; ++i) {
            if (nums[i] == nums[i-1]) continue;
            int ord = nums[i] > nums[i-1];
            if (ord != pre_ord) res++;
            pre_ord = ord;
        }
        return res;
    }
};

后记

鼠年快乐,新年献给大家的第一道题,尽量写的详细一点。

官方题解没有严谨的证明,虽然方法也是这 5 种,但是没有说清楚,不能令人信服。


   转载规则


《每日算法系列【LeetCode 376】摆动序列》 韦阳 采用 知识共享署名 4.0 国际许可协议 进行许可。
 上一篇
心有所鼠,鼠年快乐~ 心有所鼠,鼠年快乐~
关注公众号【算法码上来】,每日算法干货马上就来! 首先,新的一年祝大家鼠年快乐啦。不要问我心有所鼠到底是什么意思,我心里只有学习~ 虽然这个年发生了很多事情,可能不是很好过,但是我相信一切都会过去的。我们应该微笑着面对困难,奥里给~
2020-01-24
下一篇 
每日算法系列【LeetCode 943】最短超级串 每日算法系列【LeetCode 943】最短超级串
关注公众号【算法码上来】,每日算法干货马上就来! 题目描述给定一个字符串数组 A,找到以 A 中每个字符串作为子字符串的最短字符串。 我们可以假设 A 中没有字符串是 A 中另一个字符串的子字符串。 示例1 输入: ["alex","
2020-01-23
  目录