牛客-华为机试-入门-排序-HJ101 输入整型数组和排序标识，对其元素按照升序或降序进行排序-C++

描述

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

数据范围： 1≤n≤1000，元素大小满足 0≤val≤100000

输入描述：

第一行输入数组元素个数
第二行输入待排序的数组，每个数用空格隔开
第三行输入一个整数0或1。0代表升序排序，1代表降序排序

输出描述：

输出排好序的数字

示例1

输入：

8
1 2 4 9 3 55 64 25
0

输出：

1 2 3 4 9 25 55 64

示例2

输入：

5
1 2 3 4 5
1

输出：

5 4 3 2 1

代码

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int main(){
    int n;
    cin>>n;
    vector<int>a(n,0);
    for(int i=0;i<n;i++){
        cin>>a[i];
    }
    int flag;
    cin>>flag;    
    if(flag==0){
        sort(a.begin(),a.end());
    }
    else if(flag==1){
        sort(a.begin(),a.end());
        reverse(a.begin(),a.end());
    }
    for(int i=0;i<n-1;i++){
        cout<<a[i]<<' ';
    }
    cout<<a[n-1]<<endl;
    return 0;
}