PAT 7-40 奥运排行榜 (25 point(s))

 每年奥运会各大媒体都会公布一个排行榜，但是细心的读者发现，不同国家的排行榜略有不同。比如中国金牌总数列第一的时候，中国媒体就公布“金牌榜”；而美国的奖牌总数第一，于是美国媒体就公布“奖牌榜”。如果人口少的国家公布一个“国民人均奖牌榜”，说不定非洲的国家会成为榜魁…… 现在就请你写一个程序，对每个前来咨询的国家按照对其最有利的方式计算它的排名。

输入格式:

输入的第一行给出两个正整数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 <bits/stdc++.h>
using namespace std;

struct Nation{
    int num;             // 编号
    int ranks[4];        // 排名
    double medals[4];    // 奖牌榜
};

int main() {
    int n, m;
    cin >> n >> m;
    vector<Nation> nation(n);
    
    // 读取 奖牌 人数 并计算人均
    for(int i = 0; i < n; i++){
        int ppl;
        auto& tmp = nation[i];
        tmp.num = i;
        cin >> tmp.medals[0] >> tmp.medals[1] >> ppl;
        tmp.medals[2] = tmp.medals[0] / ppl;
        tmp.medals[3] = tmp.medals[1] / ppl;
    }
    
    for(int i = 0; i < 4; i++){
        // 不同榜单名次排序 lambda作为比较函数 []捕获i
        sort(begin(nation), end(nation), [i](const Nation x, const Nation y){
            return x.medals[i] > y.medals[i];
        });
        // j是后一个国家 j-1是前一个国家 (j = 1 j-1 = 0下标开始)
        // nation(n)开辟容器空间 元素初始化都为0 因为sort已排好序首地址元素下标必为0 故j从1开始
        // 最后输出时 *r + 1 将下标转换为实际序号
        for(int j = 1; j < n; j++){
            // 从j = 1开始按排好顺序赋值 同分同名次
            if(nation[j].medals[i] == nation[j-1].medals[i])
                nation[j].ranks[i] = nation[j-1].ranks[i];
            else nation[j].ranks[i] = j;
        }
    }
    // 还原以编号为顺序 以变输出时读取正确国家编号
    sort(begin(nation), end(nation), [](const Nation x, const Nation y){
        return x.num < y.num;
    });
    
    while(m--){
        int n;
        cin >> n;
        // 找到最大名次地址
        auto r = min_element(begin(nation[n].ranks), end(nation[n].ranks));
        // 排名 : 排名计算方式 (= 最大奖牌榜地址 - 奖牌榜首地址 + 1
        cout << *r + 1 << ":" << r - begin(nation[n].ranks) + 1 << (m == 0 ? "" : " ");
    }
}

参考代码