C++ 【贪心】求摇摆序列的最长长度

摇摆序列

题目

一个整数序列，如果两个相邻元素的差恰好正负(负正)，交替出现，则该序列被称为摇摆序列。一个小于2个元素的序列直接为摇摆序列。
例如:
序列[1, 7,4, 9, 2, 5]，相邻元素的差(6,-3,5,-7,3)，该序列为摇摆序列。
序列[1 ,4,7,2,5] [7,4,5,5]不是摇摆序列。
给一个随机序列，求这个序列满足摇摆序列定义的最长子序列的长度。
例如：
输入[1,7,4,9,2,5]，结果为6;输入[1,17,5,10,13,15,10,5,16,8]， 结果为7([1,17,10,13,10,16,8]):输入[1.2.3.4.5.6.7.8.91. 结果为2。

思路

如果在某段子序列当中，元素的大小呈连续递增或递减，即超过两个元素的连续增加或减少，该选择哪个元素，使得最终选择的子序列的摇摆长度最长？
所以，按照贪心的思想，我们只需要把增大的过程当中，最大的元素作为摇摆点，则下一个元素一定相对与这个最大元素减少；把减少的过程当中，最小的元素作为摇摆点，则下一个元素一定相对于这个最小元素增加。
按照上述的思路算出的摇摆子序列长度一定是最长的。

摇摆序列折线图形式：

而如何表示序列处于递增或者递减的状态，就需要用到状态机。
状态机从开始BEGIN时，会出现以下情况：
①相邻的元素大小增加了，则状态机进入UP状态。
②相邻的元素大小减少了，则状态机进入DOWN状态。
当BEGIN、UP、DOWN三者之间的状态发生转变时，长度加1。
状态机的形式如下图所示：

代码如下：

#include <stdio.h>
#include <vector>
class Solution {
public:
    int wiggleMaxLength(std::vector<int>& nums) {
    	if (nums.size() < 2){
	    	return nums.size();
	    }
	    static const int BEGIN = 0;
	    static const int UP = 1;
	    static const int DOWN = 2;
	    int STATE = BEGIN;
	    int max_length = 1;
    	for (int i = 1; i < nums.size(); i++){
    		switch(STATE){
	    	case BEGIN:
	    		if (nums[i-1] < nums[i]){
	    			STATE = UP;
		    		max_length++;
		    	}
		    	else if (nums[i-1] > nums[i]){
	    			STATE = DOWN;
	    			max_length++;
	    		}
	    		break;
	    	case UP:
	    		if (nums[i-1] > nums[i]){
	    			STATE = DOWN;
		    		max_length++;
		    	}
		    	break;
	    	case DOWN:
	    		if (nums[i-1] < nums[i]){
	    			STATE = UP;
		    		max_length++;
		    	}
		    	break;
		    }
	    }
    	return max_length;
    }
};

致谢

本章知识点和思路由小象学院相关视频提供，由本人学习并梳理得出，希望自己加深记忆的同时，也能给大家提供更多有关于一些算法的知识点。
你的点赞、评论、收藏就是对我最大的支持与鼓励，谢谢！