376. 擺動序列

一、問題

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

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

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

二、示例

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

三、算法思想

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

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

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

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

下面提交代碼

四、提交代碼

以下代碼的時間復(fù)雜度為O(n)，空間復(fù)雜度為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]){// 期望下一個數(shù)是小于該數(shù)的 STATE = DOWN;ans++;} else if(nums[i] < nums[i-1]){//期望下一個數(shù)是大于該數(shù)的 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ū)補(bǔ)充。

五、測試代碼

#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]){// 期望下一個數(shù)是小于該數(shù)的 STATE = DOWN;ans++;} else if(nums[i] < nums[i-1]){//期望下一個數(shù)是大于該數(shù)的 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; }

補(bǔ)充與總結(jié)

當(dāng)遇到重復(fù)的if - else 結(jié)構(gòu)的時候，還可以使用switch來替換。使得程序結(jié)構(gòu)更加清晰

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

總結(jié)

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

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。