PTA 7-40 奧運排行榜

每年奧運會各大媒體都會公佈一個排行榜，但是細心的讀者發現，不同國家的排行榜略有不同。比如中國金牌總數列第一的時候，中國媒體就公佈“金牌榜”；而美國的獎牌總數第一，於是美國媒體就公佈“獎牌榜”。如果人口少的國家公佈一個“國民人均獎牌榜”，說不定非洲的國家會成爲榜魁…… 現在就請你寫一個程序，對每個前來諮詢的國家按照對其最有利的方式計算它的排名。
輸入格式:
輸入的第一行給出兩個正整數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;
}