笔试 - 排序子序列

排序子序列

牛牛定义排序子序列为一个数组中一段连续的子序列,并且这段子序列是非递增或者非递减排序的。牛牛有一个长度为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 Test01 {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        int[] a = new int[n+1];
        for(int i = 0; i < n; i++)
            a[i] = sc.nextInt();
        //应对 1 3 1这样的序列的情况，变成 1 3 1 1 来处理。
        a[n] = a[n-1];
        n++;
        int ans = 0;
        boolean flag = false;
        int i = 1;
        while (i < n) {
            while (i < n && a[i] >= a[i-1]){
                i++;
                flag = true;
            }
            if (flag){
                ans++;
                i++;
                flag = false;
            }
            while (i < n && a[i] <= a[i-1]){
                i++;
                flag = true;
            }
            if (flag){
                ans++;
                i++;
                flag = false;
            }
        }
        System.out.println(ans);
    }
}

2020/02/27 牛客网模拟笔试编程题的第一题
自己想的测试例子有：

3
1 3 1 

5
2 2 1 3 1 或
2 2 1 2 3 或
2 2 3 1 2 或
2 2 3 2 1