华为机试-输入整型数组和排序标识，对其元素按照升序或降序进行排序（较难）

问题描述

输入整型数组和排序标识，对其元素按照升序或降序进行排序（一组测试用例可能会有多组数据）
本题有多组输入，请使用while(cin>>)处理

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
static bool cmp(int a,int b){
    return a>b;
}
int main(){
    int num;
    bool flag;
    while(cin>>num){
        vector<int> input;
        while(num--){
            int a;
            cin>>a;
            input.push_back(a);
        }
        cin>>flag;
        if(flag)  sort(input.begin(),input.end(),cmp);
        else  sort(input.begin(),input.end());
        for(int i:input)
            cout<<i<<' ';
        cout<<endl;
    }
}