【PTA数据结构 | C语言版】模拟EXCEL排序

本专栏持续输出数据结构题目集，欢迎订阅。

题目

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

输入格式:
输入的第一行包含两个正整数 n (≤10^5) 和 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 <stdio.h>
#include <stdlib.h>
#include <string.h>

// 学生记录结构体
typedef struct {
    char id[7];    // 学号（6位数字）
    char name[9];  // 姓名（不超过8位）
    int score;     // 成绩
} Student;

int c;  // 全局变量，指定排序的列号

// 比较函数：根据c的值进行不同方式的比较
int compare(const void *a, const void *b) {
    Student *s1 = (Student *)a;
    Student *s2 = (Student *)b;
    
    if (c == 1) {
        // 按学号排序
        return strcmp(s1->id, s2->id);
    } else if (c == 2) {
        // 按姓名排序，姓名相同则按学号排序
        int name_cmp = strcmp(s1->name, s2->name);
        if (name_cmp != 0) {
            return name_cmp;
        } else {
            return strcmp(s1->id, s2->id);
        }
    } else {  // c == 3
        // 按成绩排序，成绩相同则按学号排序
        if (s1->score != s2->score) {
            return s1->score - s2->score;
        } else {
            return strcmp(s1->id, s2->id);
        }
    }
}

int main() {
    int n;
    scanf("%d %d", &n, &c);
    
    // 分配内存存储学生记录
    Student *students = (Student *)malloc(n * sizeof(Student));
    
    // 读取学生记录
    for (int i = 0; i < n; i++) {
        scanf("%s %s %d", students[i].id, students[i].name, &students[i].score);
    }
    
    // 排序
    qsort(students, n, sizeof(Student), compare);
    
    // 输出排序结果
    for (int i = 0; i < n; i++) {
        printf("%s %s %d\n", students[i].id, students[i].name, students[i].score);
    }
    
    return 0;
}