小苯的排序疑惑（Java）

题目描述

小苯有一个长度为 n 的数组 a，他可以对 a 进行至多一次以下操作：

选择一段区间 [l,r]，满足 (1≤l≤r≤n)，且区间长度严格小于 n，将数组 a 的 [l,r] 这段区间按非降序排序。

换句话说，操作执行完后，区间中的值将满足：a[l]≤a[l+1]≤a[l+2]≤...≤a[r]。

现在小苯想知道能否通过执行最多一次操作使得数组 a 按非降序排列。

输入描述:

本题有多组测试用例。 第一行一个正整数 t(1≤t≤10^4) 表示测试用例的组数。 每组数据第一行一个正整数 n(1≤n≤2×10^5)，表示数组的长度。 每组数据第二行 n 个正整数 ai(1≤ai≤10^9)，表示数组 a 的元素。

输入保证所有测试用例中的 n 总和不超过 2×10^5。

输出描述

输出包含 t 行，表示每组用例的答案。 如果可以使 a 有序，输出 "YES"，否则输出 "NO"（输出不含双引号）。

示例1

输入

3

4

1 3 2 4

3

3 5 2

1

1

输出

YES

NO

YES

说明

第一个测试用例，选择排序 [2,3] 这个区间，数组变为 [1,2,3,4]。因此输出YES。 第二个测试用例，无法选择合法的区间使得数组有序。 第三个测试用例，不需要选择区间，数组本身就有序。

备注

数组按非降序排列：指数组每一项值都大于等于前一项值（如果前一项存在的话）。 形式化的： 对于任意 1≤i≤n−1都有 ai≤ai+1。 特别的，我们认为长度恰好为 1 的数组也是非降序的。

思路

选择的区间长度严格意义上要小于数组长度n，所以我们能选择操作的区间长度最多只能是n - 1，这意味着数组的a[0]或者a[n - 1]，我们无法去操作它。在这种情况下，要保证操作执行完后，数组a是非降序排列，只有当a[0]是最小值，或a[n - 1]是最大值时，才能实现。

代码

import java.util.Scanner;
import java.util.Arrays;
public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int t = sc.nextInt() ; //测试数据组数
        for (int i = 0; i < t; i++) {
            int n = sc.nextInt() ; //数组长度
            int []a = new int [n] ;
            for (int j = 0; j < n; j++) {
                a[j] = sc.nextInt() ;
            }
            int min = a[0] ; int max = a[n - 1] ;排序前，假设a[0]是最小，a[n - 1]是最大
            Arrays.sort(a) ;//将数组排序(从小到大)
            if(a[0] == min || a[n - 1]== max){  //排序后，还是a[0]最小或a[n-1]最大
                System.out.println("YES") ;
            }else{
                System.out.println("NO") ;
            }
        }
    }
}