376. 擺動序列

一、問題

如果連續(xù)數字之間的差嚴格地在正數和負數之間交替，則數字序列稱為擺動序列。第一個差（如果存在的話）可能是正數或負數。少于兩個元素的序列也是擺動序列。

例如， [1,7,4,9,2,5] 是一個擺動序列，因為差值 (6,-3,5,-7,3) 是正負交替出現(xiàn)的。相反, [1,4,7,2,5] 和 [1,7,4,5,5] 不是擺動序列，第一個序列是因為它的前兩個差值都是正數，第二個序列是因為它的最后一個差值為零。

給定一個整數序列，返回作為擺動序列的最長子序列的長度。 通過從原始序列中刪除一些（也可以不刪除）元素來獲得子序列，剩下的元素保持其原始順序。

二、示例

示例 1:

輸入: [1,7,4,9,2,5]
輸出: 6
解釋: 整個序列均為擺動序列。
示例 2:

示例 2：

輸入: [1,17,5,10,13,15,10,5,16,8]
輸出: 7
解釋: 這個序列包含幾個長度為 7 擺動序列，其中一個可為[1,17,10,13,10,16,8]。
示例 3:

示例 3：

輸入: [1,2,3,4,5,6,7,8,9]
輸出: 2

三、算法思想

這道題其實思路十分清晰，我們以 [1,17,5,10,13,15,10,5,16,8]序列為例作圖。

有該圖例就可以清晰知道我們具體的邏輯結構和有效節(jié)點

首先，a_i若比下一個節(jié)點a_i+1小，那我們期望的下下一個有效節(jié)點就是a_i+2比a_i+1大，依次反復

其次，當遇到連續(xù)遞增或遞減時，去遞增最大值或遞減最小值（貪心）即可

下面提交代碼

四、提交代碼

以下代碼的時間復雜度為O(n)，空間復雜度為O(n)

int wiggleMaxLength(std::vector<int>& nums) {// 最終返回值 if (nums.size() < 2) return nums.size();int ans = 1;const int UP = 1;const int DOWN = 2;const int BEGIN = 0;int STATE = BEGIN;for (int i = 1; i < nums.size(); i++){switch(STATE){case BEGIN:if(nums[i] > nums[i-1]){// 期望下一個數是小于該數的 STATE = DOWN;ans++;} else if(nums[i] < nums[i-1]){//期望下一個數是大于該數的 STATE = UP;ans++;}break;case UP:if(nums[i] > nums[i-1]){STATE = DOWN;ans++;}break;case DOWN:if(nums[i] < nums[i-1]){STATE = UP;ans++;}break;}}return ans; }

若有其他解法，歡迎評論區(qū)補充。

五、測試代碼

#include <stdio.h> #include <vector>int wiggleMaxLength(std::vector<int>& nums) {// 最終返回值 if (nums.size() < 2) return nums.size();int ans = 1;const int UP = 1;const int DOWN = 2;const int BEGIN = 0;int STATE = BEGIN;for (int i = 1; i < nums.size(); i++){switch(STATE){case BEGIN:if(nums[i] > nums[i-1]){// 期望下一個數是小于該數的 STATE = DOWN;ans++;} else if(nums[i] < nums[i-1]){//期望下一個數是大于該數的 STATE = UP;ans++;}break;case UP:if(nums[i] > nums[i-1]){STATE = DOWN;ans++;}break;case DOWN:if(nums[i] < nums[i-1]){STATE = UP;ans++;}break;}}return ans; }int main(){std::vector<int> nums;nums.push_back(1);nums.push_back(7);nums.push_back(4);nums.push_back(9);nums.push_back(2);nums.push_back(5);int ans = wiggleMaxLength(nums);printf("%d",ans);return 0; }

補充與總結

當遇到重復的if - else 結構的時候，還可以使用switch來替換。使得程序結構更加清晰

來源：力扣（LeetCode）376. 擺動序列

總結

以上是生活随笔為你收集整理的【LeetCode】376. 摆动序列（图解）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。