输入整型数组和排序标识，对其元素按照升序或降序进行排序

题目描述：

输入整型数组和排序标识，对其元素按照升序或降序进行排序

排序标识：0表示按升序，1表示按降序

输入描述：

输入需要输入的整型数个数

输出描述：

输出排好序的数字

import java.util.Arrays;
import java.util.Scanner;

public class Main
{
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        while (scanner.hasNext())
        {
            int n = scanner.nextInt();
            int[] arr = new int[n];
            for (int i = 0; i < n; i++)
                arr[i] = scanner.nextInt();
            int flag = scanner.nextInt();
            Arrays.sort(arr);
            if (flag == 0)
            {
                for (int i = 0; i < arr.length - 1; i++)
                    System.out.print(arr[i] + " ");
                System.out.println(arr[arr.length - 1]);
            }
            else 
            {
                for (int i = arr.length - 1; i >= 1; i--)
                    System.out.print(arr[i] + " ");
                System.out.println(arr[0]);
            }
        }
    }
}