PAT-A 1028. List Sorting (25)

题目链接在此。

题意

给出N条学学生信息，学生信息包含三项信息（学号、姓名、分数），之后给出一个整数C，按照C的情况进行如下排序：

1. C==1，按照学号排升序
2. C==2，按照姓名排升序
3. C==3，按照分数排升序
   如果姓名和分数相同，按学号从小到大排序。

AC代码

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>

using namespace std;

struct INFO{
    int id;
    char name[9];
    int grade;
}stu[100001];

//cmp for C=1
bool cmp1(INFO a, INFO b){
    return a.id < b.id;
} 

//cmp for C=2
bool cmp2(INFO a, INFO b){
    if(strcmp(a.name, b.name) != 0) return (strcmp(a.name, b.name)<0); //name作比较时注意用strcmp函数，不能直接== ，否则有一组数据过不了 
    else return a.id < b.id; 
}

//cmp for C=3
bool cmp3(INFO a, INFO b){
    if(a.grade != b.grade) return a.grade < b.grade;
    else return a.id < b.id;
}

int main(){

    int N,C;
    scanf("%d %d",&N,&C); 

    for(int i = 0 ; i < N; i++){
        scanf("%d %s %d",&stu[i].id,&stu[i].name,&stu[i].grade);
    }

    switch(C){
        case 1: {
            sort(stu,stu+N,cmp1);
            break;
        }
        case 2:{
            sort(stu,stu+N,cmp2);
            break;
        }
        case 3:{
            sort(stu,stu+N,cmp3);
            break;
        }
    } 

    for(int i = 0; i < N; i++){
        printf("%06d %s %d\n",stu[i].id,stu[i].name,stu[i].grade);
    }

    return 0;
}