PTA basic 1055 集体照 (25 分) c++语言实现(g++)

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
using namespace std;
class student {
public:
    string name;
    int tall;
    student()=default;
    student(string name_,int tall_):name{name_},tall{tall_}{};
};
void reOrder(vector<student>&templist){//按照要求,把最高的放 行中间,然后按照先左后右的顺序依次放入次高的
    int size=templist.size();
    vector<student> temp(size);
    int mid=size/2,left{mid-1},right={mid+1},count{1};
    temp[mid]=templist[0];
    while(count!=size){
        if(count%2){//先左
            if(left>=0){
                temp[left--]=templist[count++];
            }else if(right<size){//左不存在放右边,因为第一个元素是放置在中间的,所以只需要处理一次边界就可以放完一排的全部元素
                temp[right++]=templist[count++];
            }
        }else{//后右
            if(right<size){
                temp[right++]=templist[count++];
            }else if(left>=0){//右不存在放左边
                temp[left--]=templist[count++];
            }
        }
    }
    templist=temp;
}

bool compare(student l,student r){//快排的比较规则,身高大的 或者 字母序小的 元素 是 大元素, 小元素在前,大元素靠后
    bool flag=0;
    if(l.tall!=r.tall){//身高不同时
        flag=l.tall<r.tall;//左边身高 小于 右边时,右边的是大元素
    }else{//同身高比较name字符串
        flag=l.name.compare(r.name)>0;// l.name字符串 逐个跟 r.name字符串比较,
        //string.compare(argument)的返回值, string字符大于参数argument的字符时,返回1,小于返回-1,等于返回0
        //如果l的字符 大于 r的字符 , 既返回值为1时,   右边是小字母,右边是大元素
    }
    return flag;
}
int main(){
    int n,k,tempTall;
    int row,lastRow;
    string tempName;
    vector<vector<student>> list;
    vector<student>tempList;
    vector<student>tempRow;
    cin >> n>> k;
    if(k>10)k=10;//排数,最多10排
    row=n/k;//除最后一排外,每排站多少人
    lastRow=row+n%k;//剩下的人数
    
    for(int i=0;i<n;i++){
        cin >> tempName >> tempTall;
        student tempStudent{tempName,tempTall};
        tempList.push_back(tempStudent);
    }
    
    //按身高从小到大排序,同身高的按名字的字母从大到小排序
    sort(tempList.begin(),tempList.end(),compare);
    
    //拆分成多排,先把最后一排的元素 加入到二维数组中
    for(int i=0;i<lastRow;i++){
        tempRow.push_back(tempList.back());
        tempList.pop_back();
    }
    list.push_back(tempRow);
    tempRow.clear();
    for(int i=k-2;i>=0;i--){//再把剩下的每一排,按照从 倒数第二排 到 第一排 的顺序加入到二维数组中
        for(int j=row-1;j>=0;j--){
            tempRow.push_back(tempList[i*row+j]);
        }
        list.push_back(tempRow);
        tempRow.clear();
    }
    
    //每一排都重排序
    for(int i=0;i<list.size();i++){
        reOrder(list[i]);
    }
    
    //按序输出
    for(int i=0;i<list.size();i++){
        int j;
        for(j=0;j<list[i].size()-1;j++){
            cout << list[i][j].name<<" ";
        }
        cout <<list[i][j].name<<endl;//每一排末尾单独输出
    }
    return 0;
}