牛牛的数列

牛定义排序子序列为一个数组中一段连续的子序列,并且这段子序列是非递增或者非递减排序的。牛牛有一个长度为n的整数数组A,他现在有一个任务是把数组A分为若干段排序子序列,牛牛想知道他最少可以把这个数组分为几段排序子序列.
如样例所示,牛牛可以把数组A划分为[1,2,3]和[2,2,1]两个排序子序列,至少需要划分为2个排序子序列,所以输出2 

输入描述:
输入的第一行为一个正整数n(1 ≤ n ≤ 10^5) 
第二行包括n个整数A_i(1 ≤ A_i ≤ 10^9),表示数组A的每个数字。
输出描述:
输出一个整数表示牛牛可以将A最少划分为多少段排序子序列
输入例子:
6
1 2 3 2 2 1
输出例子:
2

这道题的基本思想:定义两个指针begin、end,分别指向数组的第一个元素和后一个元素,然后,分三种情况讨论:
1、str[begin]==str[end],则begin++,end++,然后跳出循环
2、str[begin]<str[end],则进入递减序列的判断
3、str[begin]>str[end],则进入递增序列的判断

测试用例:
2 3 2 3 2 3
1 1 1 1 4 5
6 7 5 3 4 4 2
#include<bits/stdc++.h>
using namespace std;

int main()
{
    int n;
    cin>>n;
    vector<int> str(n);
    int count=0;
    int begin=0;
    int end = 1;
    for(int i=0;i<n;i++)
    {
        cin>>str[i];
    }
    if(n==1)
    {
        cout<<"1"<<endl;
        return 0;
    }
    while(end<n)
    {

        if(str[begin]==str[end])
        {
            begin=end;
            end++;
            continue;
        }
        if(str[begin]<str[end]&&end<n)
        {
            count++;
            while(str[begin]<=str[end]&&end<n)
            {
                begin=end;
                end++;
            }
            if(end<n-1)
            {
                begin=end;
                end++;
            }
        }

        if(str[begin]>str[end]&&end<n)
        {
            count++;
            while(str[begin]>=str[end]&&end<n)
            {
                begin=end;
                end++;
            }
            if(end<n-1)
            {
                begin=end;
                end++;
            }
        }
    }
    cout<<count<<endl;
    return 0;
}