ACM: EXCEL排序

一、题目大意

题目描述：
Excel可以对一组纪录按任意指定列排序。现请你编写程序实现类似功能。
对每个测试用例，首先输出1行“Case i:”，其中 i 是测试用例的编号（从1开始）。随后在 N 行中输出按要求排序后的结果，即：当 C=1 时，按学号递增排序；当 C=2时，按姓名的非递减字典序排序；当 C=3时，按成绩的非递减排序。当若干学生具有相同姓名或者相同成绩时，则按他们的学号递增排序。

输入：
测试输入包含若干测试用例。每个测试用例的第1行包含两个整数 N (N<=100000) 和 C，其中 N 是纪录的条数，C 是指定排序的列号。以下有N行，每行包含一条学生纪录。每条学生纪录由学号（6位数字，同组测试中没有重复的学号）、姓名（不超过8位且不包含空格的字符串）、成绩（闭区间[0, 100]内的整数）组成，每个项目间用1个空格隔开。当读到 N=0 时，全部输入结束，相应的结果不要输出。

输出：
对每个测试用例，首先输出1行“Case i:”，其中 i 是测试用例的编号（从1开始）。随后在 N 行中输出按要求排序后的结果，即：当 C=1 时，按学号递增排序；当 C=2时，按姓名的非递减字典序排序；当 C=3
时，按成绩的非递减排序。当若干学生具有相同姓名或者相同成绩时，则按他们的学号递增排序。

样例输入

3 1
000007 James 85
000010 Amy 90
000001 Zoe 60
4 2
000007 James 85
000010 Amy 90
000001 Zoe 60
000002 James 98
4 3
000007 James 85
000010 Amy 90
000001 Zoe 60
000002 James 90
0 0

样例输出

Case 1:
000001 Zoe 60
000007 James 85
000010 Amy 90
Case 2:
000010 Amy 90
000002 James 98
000007 James 85
000001 Zoe 60
Case 3:
000001 Zoe 60
000007 James 85
000002 James 90
000010 Amy 90

二、解题思路

分情况重载<号

三、代码

#include<iostream>
#include<algorithm>
using namespace std;

const int MAX = 100005;
int c, N;
struct Node
{
    string id;
    string name;
    int grad;
    bool operator <(const Node&A)const
    {
        if(c==1)
            return id < A.id;
        else if(c == 2)
        {
            if(name != A.name)
                return name < A.name;
            return id < A.id;
        }
        else
        {
            if(grad != A.grad)
                return grad < A.grad;
            return id < A.id;
        }
    }
}a[MAX];

int main()
{
    int cnt = 1;
    while(cin >> N >> c)
    {
        if(N + c == 0)
            break;
        for(int i=0; i<N; i++)
            cin >> a[i].id >> a[i].name >> a[i].grad;
        sort(a, a+N);

        cout << "Case " << cnt++ << ":"<< endl;
        for(int i=0; i<N; i++)
            cout << a[i].id << " " << a[i].name << " " << a[i].grad << endl;
    }
    return 0;
}