排序子序列-Java

题目：

牛牛定义排序子序列为一个数组中一段连续的子序列,并且这段子序列是非递增或者非递减排序的。牛牛有一个长度为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

实现代码：

import  java.util.Scanner;
public class Demo{
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        while (sc.hasNextInt()){
            int n = sc.nextInt();
            int[] a = new int[n];
            for(int i = 0; i < n; i++){
                a[i] = sc.nextInt();
            }
            int flag = 0;//这里0表示不增不减，1表示递增，-1表示递减
            int count = 1;//计数，默认从1当前序列开始
            for (int i = 0; i < n; i++) {
                if (a[i-1]<a[i]){
                        if (flag == 0){//初值
                            flag = 1;//递增
                        }
                        if (flag == -1) {//从递增序列转为递减序列
                            flag = 0;//重制，避免重复分组时重复利用同一个（如：{1,2,1,2}错误分为{1,2}{2,1}{1,2}三组，正确情况应是{1,2}{1,2}）
                            count++;
                        }
                }else if (a[i-1]>a[i]){
                    if (flag == 0){
                        flag = -1;//递减
                    }
                    if (flag == 1) {//从递增序列转到递减
                        flag = 0;//重置
                        count++;
                    }
                }
            }
            System.out.println(count);
        }
        sc.close();
    }
}