c++ 摇摆序列 (贪心+状态自动机)

一个整数序列，如果相邻两个元素差是正负交替，则该序列为摇摆序列，
小于2个元素并且不相等，序列直接是摇摆序列
如
序列[1,7,4,9,2,5] 相邻元素差(6,-3,5,-7,3)，是摇摆序列
序列[1,4,7,2,5] 相邻元素差(3,3,-5,3)不是摇摆序列
给一个随机序列，求满足 摇摆序列 定义的 最长子序列长度
例如
[1,7,4,9,2,5] (6,-3,5,-7,3) 结果为6 
[1,17,5,10,13,15,10,5,16,8] 最长子序列-> [1,17,10,13,10,16,8] 结果7 
[1,2,3,4,5,6,7,8,9] 结果为2 

以 [1,17,5,10,13,15,10,5,16,8]为例

 

 

 10 13 15是连续递增，要弄摇摆序列，这三个数字肯定不能同时出现，要从这里面选一个，选择最大的一个数字(贪心)，才可能与原始第七位组成摇摆序列，

 

 在递增或者递减序列时候，只保留首和尾元素（中间的元素就不要了），这样更有可能使得尾部的后一个元素成为摇摆序列

 

 当发生折现时摇摆序列max_len++，当发生递增和递减时，只加一次，相当于越过中间的值
上面的三角转换图（状态机），红色就是++时候状态转移时，暗蓝色就是状态没有转换
蓝色代码其实什么都没有做，所以就不用写，根据判断条件只有发生折线时候才会转移状态进行++操作，当前状态递增和递减时候就没有转移，，根据状态机的线就可以写出每个状态有几个if

#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
using namespace std;
int wig(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;//摇摆序列最大长度至少为1,0个和1个元素就返回了，现在至少两个元素 
	
	/*
	 从下标1（第二个数字）开始，因为要和前一个元素比较，
	 如果是下标0开始就会越界，而且第一个摇摆序列长度默认为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;
		} //switch 
	} //for 
	return max_length;	
} 
int main(){
	//[1,17,5,10,13,15,10,5,16,8] - [1,17,10,13,10,16,8] 结果7
	vector<int> v;
	v.push_back(1);v.push_back(17);v.push_back(5);
	v.push_back(10);v.push_back(13);v.push_back(15);v.push_back(10);
	v.push_back(5);v.push_back(16);v.push_back(8);
	cout<<wig(v);
}