PTA:模拟EXCEL排序

Excel可以对一组纪录按任意指定列排序。现请编写程序实现类似功能。

输入格式:

输入的第一行包含两个正整数 n (≤105) 和 c，其中 n 是纪录的条数，c 是指定排序的列号。之后有 n 行，每行包含一条学生纪录。每条学生纪录由学号（6 位数字，保证没有重复的学号）、姓名（不超过 8 位且不包含空格的字符串）、成绩（[0, 100] 内的整数）组成，相邻属性用 1 个空格隔开。

输出格式:

在 n 行中输出按要求排序后的结果，即：当 c=1 时，按学号递增排序；当 c=2 时，按姓名的非递减字典序排序；当 c=3 时，按成绩的非递减排序。当若干学生具有相同姓名或者相同成绩时，则按他们的学号递增排序。

输入样例:

3 1
000007 James 85
000010 Amy 90
000001 Zoe 60

输出样例:

000001 Zoe 60
000007 James 85
000010 Amy 90

 

代码如下：

#include<iostream>
#include<vector>
#include<string>
#include<algorithm>
using namespace std;struct Student{string number;string name;int grade;
};bool compareByNumber(const Student& a,const Student& b)
{return a.number<b.number;
}bool compareByName(const Student& a,const Student& b)
{if(a.name==b.name){return a.number<b.number;}return a.name<b.name;
}bool compareByGrade(const Student& a,const Student& b)
{if(a.grade==b.grade){return a.number<b.number;}return a.grade<b.grade;
}void sortStudents(vector<Student>& students,int c)
{if(c==1){sort(students.begin(),students.end(),compareByNumber);}else if(c==2){sort(students.begin(),students.end(),compareByName);}else if(c==3){sort(students.begin(),students.end(),compareByGrade);}
}int main()
{int n,c;cin>>n>>c;vector<Student> students(n);for(int i=0;i<n;i++){cin>>students[i].number>>students[i].name>>students[i].grade;}sortStudents(students,c);for(Student& student:students){cout<<student.number<<" "<<student.name<<" "<<student.grade<<endl;}return 0;
}

学习心得 ：

1、StudentA->name - StudentB->name 和 StudentA->number - StudentB->number 这种操作是错误的。name 和 number 是 std::string 类型，不能直接进行减法运算。应该使用字符串的比较函数，如 compare 或者直接使用比较运算符（<、> 等）。