浙大 PAT 甲级 1055 The World's Richest 归并排序

这题的母题是找前k个最大数，但是有了年龄限制，并且年龄与要排序的财富实际大小没有关系。年龄是整数且有限。

这些事实不得不让人想到归并排序。前面也有一道题用到了归并排序，如果是按顺序做下来的，这题很容易想到归并排序。

用结构体P表示一个人，保存其姓名和财富值。在读取每个人的信息时，按年龄来存放，一个年龄对应一个vector<P>。这样得到了一个以vector<P>为元素类型的，大小为201（题目中限制年龄范围为1-200）的数组。

对数组的每个vector元素按要求排序，然后对每个要求进行归并排序找到前k个最大数即可。

如果对归并排序不太了解，建议查看《数据结构与算法分析》，其中对归并排序做了详细讲解。归并排序也是机试中常考的排序方法。

这题有些小细节。一个都没有的时候输出None，如果满足条件的人数小于k，那么输出满足条件的这些人即可，不一定非得找到前k个，这时候后面不能输出None。

还要使用一个数组，来记录归并排序时每个vector当前排序到第几个元素。

下面贴出AC代码

#include<stdio.h>
#include<vector>
#include<algorithm>
#include<string>
#include<string.h>
#include<memory.h>
using namespace std;

struct P
{
    char name[9];
    int worth;
};

vector<P> people[201];
int indexOfUnsorted[201]; // 归并排序，记录数组中每个vector中未排序的第一个元素的下标

bool cmp(P a, P b)
{
    if (a.worth != b.worth)
    {
        return (a.worth > b.worth);
    }
    else
    {
        return (strcmp(a.name, b.name) < 0);
    }
}

int main()
{
    int N, K;
    scanf("%d%d", &N, &K);
    for (int i = 0; i < N; i++)
    {
        char name[9];
        int age, worth;
        scanf("%s%d%d", name, &age, &worth);
        P p;
        strcpy(p.name, name);
        p.worth = worth;
        people[age].push_back(p);
    }
    for (int i = 1; i < 201; i++)
    {
        sort(people[i].begin(), people[i].end(), cmp);
    }
    for (int i = 0; i < K; i++)
    {
        memset(indexOfUnsorted, 0, sizeof(int) * 201);
        int M, Amin, Amax;
        bool flag = false;
        scanf("%d%d%d", &M, &Amin, &Amax);
        printf("Case #%d:\n", i + 1);

        for (int j = 0; j < M; j++)
        {
            int bestAge = 0;
            int bestWorth = -1000000;
            for (int k = Amin; k <= Amax; k++)
            {
                if (indexOfUnsorted[k]<people[k].size() && people[k][indexOfUnsorted[k]].worth > bestWorth)
                {
                    bestAge = k;
                    bestWorth = people[k][indexOfUnsorted[k]].worth;
                    flag = true;
                }
            }
            if (flag && bestAge > 0)
            {
                printf("%s %d %d\n", people[bestAge][indexOfUnsorted[bestAge]].name, bestAge, bestWorth);
                indexOfUnsorted[bestAge]++;
            }
            else if (flag == false)
            {
                printf("None\n");
                break;
            }
        }
    }
    return 0;
}