PTA 7-37 模擬EXCEL排序（自定義排序函數）

本題考點：

• 自定義排序函數

Excel可以對一組紀錄按任意指定列排序。現請編寫程序實現類似功能。 輸入格式: 輸入的第一行包含兩個正整數N(≤10⁵) 和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 <algorithm>
#include <cstring>
using namespace std;

#define MAXN 100010

int N, C;

struct Stu
{
    int id;
    char name[10];
    int grade;
} stus[MAXN];

bool cmp1(Stu a, Stu b)
{
    return a.id < b.id;
}

bool cmp2(Stu a, Stu b)
{
    if (strcmp(a.name, b.name) == 0)
        return a.id < b.id;
    else
        return strcmp(a.name, b.name) < 0;
}

bool cmp3(Stu a, Stu b)
{
    if (a.grade == b.grade)
        return a.id < b.id;
    else
        return a.grade < b.grade;
}

int main()
{
    freopen("data.txt", "r", stdin);
    scanf("%d%d", &N, &C);
    getchar();
    for (int i = 0; i < N; i++)
    {
        scanf("%d%s%d", &stus[i].id, &stus[i].name, &stus[i].grade);
    }
    if (C == 1)
    {
        sort(stus, stus + N, cmp1);
    }
    else if (C == 2)
    {
        sort(stus, stus + N, cmp2);
    }
    else if (C == 3)
    {
        sort(stus, stus + N, cmp3);
    }
    for (int i = 0; i < N; i++)
    {
        if(i != N - 1)
            printf("%06d %s %d\n", stus[i].id, stus[i].name, stus[i].grade);
        else 
            printf("%06d %s %d", stus[i].id, stus[i].name, stus[i].grade);
    }
    return 0;
}