PTA 7-40 奥运排行榜

542 阅读4分钟

每年奥运会各大媒体都会公布一个排行榜,但是细心的读者发现,不同国家的排行榜略有不同。比如中国金牌总数列第一的时候,中国媒体就公布“金牌榜”;而美国的奖牌总数第一,于是美国媒体就公布“奖牌榜”。如果人口少的国家公布一个“国民人均奖牌榜”,说不定非洲的国家会成为榜魁…… 现在就请你写一个程序,对每个前来咨询的国家按照对其最有利的方式计算它的排名。
输入格式:
输入的第一行给出两个正整数N和M(≤224,因为世界上共有224个国家和地区),分别是参与排名的国家和地区的总个数、以及前来咨询的国家的个数。为简单起见,我们把国家从0 ~ N−1编号。之后有N行输入,第i行给出编号为i−1的国家的金牌数、奖牌数、国民人口数(单位为百万),数字均为[0,1000]区间内的整数,用空格分隔。最后面一行给出M个前来咨询的国家的编号,用空格分隔。
输出格式:
在一行里顺序输出前来咨询的国家的排名:计算方式编号。其排名按照对该国家最有利的方式计算;计算方式编号为:金牌榜=1,奖牌榜=2,国民人均金牌榜=3,国民人均奖牌榜=4。输出间以空格分隔,输出结尾不能有多余空格。
若某国在不同排名方式下有相同名次,则输出编号最小的计算方式。
输入样例:
4 4
51 100 1000
36 110 300
6 14 32
5 18 40
0 1 2 3
输出样例:
1:1 1:2 1:3 1:4

这道题是一个排序题的应用,难度在于如何保存每个国家和地区的每种情况下的排名,然后选择一个最好的排名。

其次要注意的是,多个相同的分数的排名应该是一样的。

注意好这两点就可以编写代码来实现了。

完整代码实现如下:

/* 
    奥运排行榜
 */
#include <iostream>
#include <algorithm>
using namespace std;

#define maxn 225

struct Goal
{
    int id, gold, medal, population;
    double averageGold, averageMedal;
} ranks[maxn];

int N, M;          // 参与排名的国家总数(0~N-1),前来咨询的国家总数
int country[maxn]; // 最后参与排名的国家

bool cmpGold(Goal a, Goal b)
{
    return a.gold > b.gold;
}

bool cmpMedal(Goal a, Goal b)
{
    return a.medal > b.medal;
}

bool cmpAverGold(Goal a, Goal b)
{
    return a.averageGold > b.averageGold;
}

bool cmpAverMedal(Goal a, Goal b)
{
    return a.averageMedal > b.averageMedal;
}

int main()
{
    scanf("%d%d", &N, &M);
    for (int i = 0; i < N; i++)
    {
        scanf("%d%d%d", &ranks[i].gold, &ranks[i].medal, &ranks[i].population);
        ranks[i].averageGold = ranks[i].gold * 1.0 / ranks[i].population;
        ranks[i].averageMedal = ranks[i].medal * 1.0 / ranks[i].population;
        ranks[i].id = i;
    }
    for (int i = 0; i < M; i++)
    {
        scanf("%d", &country[i]);
    }

    int rank1[maxn], rank2[maxn], rank3[maxn], rank4[maxn];

    // 1. 根据金牌数排名
    sort(ranks, ranks + N, cmpGold);
    int level = 1;
    for (int i = 0; i < N; i++)
    {
        if (i == 0)
        {
            rank1[ranks[i].id] = level;
        }
        else
        {   // 如果当前金牌数和上一个的金牌数一样,那么就和上一个的排名一样
            if (ranks[i].gold == ranks[i - 1].gold)
                rank1[ranks[i].id] = rank1[ranks[i - 1].id];
            else
                rank1[ranks[i].id] = level;
        }
        level++;
    }

    // 2. 根据奖牌数排名
    level = 1;
    sort(ranks, ranks + N, cmpMedal);
    for (int i = 0; i < N; i++)
    {

        if (i == 0)
        {
            rank2[ranks[i].id] = level;
        }
        else
        {   // 如果当前金牌数和上一个的金牌数一样,那么就和上一个的排名一样
            if (ranks[i].medal == ranks[i - 1].medal)
                rank2[ranks[i].id] = rank2[ranks[i - 1].id];
            else
                rank2[ranks[i].id] = level;
        }
        level++;
    }

    // 3. 根据人均金牌数排名
    level = 1;
    sort(ranks, ranks + N, cmpAverGold);
    for (int i = 0; i < N; i++)
    {
        if (i == 0)
        {
            rank3[ranks[i].id] = level;
        }
        else
        {   // 如果当前金牌数和上一个的金牌数一样,那么就和上一个的排名一样
            if (ranks[i].averageGold == ranks[i - 1].averageGold)
                rank3[ranks[i].id] = rank3[ranks[i - 1].id];
            else
                rank3[ranks[i].id] = level;
        }
        level++;
    }

    // 4.根据人均奖牌数排名
    level = 1;
    sort(ranks, ranks + N, cmpAverMedal);
    for (int i = 0; i < N; i++)
    {

        if (i == 0)
        {
            rank4[ranks[i].id] = level;
        }
        else
        {   // 如果当前金牌数和上一个的金牌数一样,那么就和上一个的排名一样
            if (ranks[i].averageMedal == ranks[i - 1].averageMedal)
                rank4[ranks[i].id] = rank4[ranks[i - 1].id];
            else
                rank4[ranks[i].id] = level;
        }
        level++;
    }

    int rankType = 0;   // 记录方案
    int bestRank = N + 1;   // 记录排名
    for (int i = 0; i < M; i++)
    {
        rankType = 0;
        bestRank = N + 1;
        if (rank1[country[i]] < bestRank)
        {
            rankType = 1;
            bestRank = rank1[country[i]];
        }
        if(rank2[country[i]] < bestRank)
        {
            rankType = 2;
            bestRank = rank2[country[i]];
        }
        if(rank3[country[i]] < bestRank)
        {
            rankType = 3;
            bestRank = rank3[country[i]];
        }
        if(rank4[country[i]] < bestRank)
        {
            rankType = 4;
            bestRank = rank4[country[i]];
        }
        if(i != 0)
            printf(" ");
        printf("%d:%d", bestRank, rankType);
        }
    return 0;
}