DreamJudge-1310-奧運排序問題（精華）

DawnTraveler發表於2024-06-15
排序

1.題目描述

Time Limit: 1000 ms
Memory Limit: 256 mb

按要求，給國家進行排名。

輸入輸出格式

輸入描述:

有多組資料。
第一行給出國家數N，要求排名的國家數M，國家號從0到N-1。
第二行開始的N行給定國家或地區的奧運金牌數，獎牌數，人口數（百萬）。
接下來一行給出M個國家號。

輸出描述:

排序有4種方式: 金牌總數 獎牌總數 金牌人口比例 獎牌人口比例
對每個國家給出最佳排名排名方式 和 最終排名
格式為: 排名:排名方式
如果有相同的最終排名，則輸出排名方式最小的那種排名，對於排名方式，金牌總數 < 獎牌總數 < 金牌人口比例 < 獎牌人口比例
如果有並列排名的情況，即如果出現金牌總數為 100,90,90,80.則排名為1,2,2,4.
每組資料後加一個空行。

輸入輸出樣例

輸入樣例#:

4 4
4 8 1
6 6 2
4 8 2
2 12 4
0 1 2 3
4 2
8 10 1
8 11 2
8 12 3
8 13 4
0 3

輸出樣例#:

1:3
1:1
2:1
1:2

1:1
1:1

題目來源
浙江大學機試題

2.題解

2.1

思路

1.這裡由於存在多種判斷規則，所以巧妙使用函式指標----函式型別（*函式名）(引數列表)，增強程式的相容性！
vector<int> getRankings(const vector<Country> &countries, bool(*compare)(const Country&, const Country&)) {...}

2.在其中根據規則對於國家進行排序，然後如何知道原有順序國家的依次排名呢？這裡我們儲存id欄位，即使排序後順序改變，也可以快速定位到原有順序了。

3.這裡要求我們如果有相同的最終排名，則輸出排名方式最小的那種排名，對於排名方式，金牌總數 < 獎牌總數 < 金牌人口比例 < 獎牌人口比例
而sort排序對於pair型別比較，首先比較第一個元素，然後比較第二個元素；
也就是在先比較排名(first)後，在比較排名方式先後（second）即可

vector<pair<int, int>> rankings = {
    {goldRankings[countryId], 1},
    {medalRankings[countryId], 2},
    {goldPerCapitaRankings[countryId], 3},
    {medalPerCapitaRankings[countryId], 4}
};
sort(rankings.begin(), rankings.end()); // 對於pair型別比較，首先比較第一個元素，然後比較第二個元素

程式碼

#include <bits/stdc++.h>
using namespace std;

struct Country {
    int id;
    int gold;
    int medal;
    int population;
};

bool compareByGold(const Country &a, const Country &b) {
    return a.gold > b.gold;
}

bool compareByMedal(const Country &a, const Country &b) {
    return a.medal > b.medal;
}

bool compareByGoldPerCapita(const Country &a, const Country &b) {
//    return static_cast<double>(a.gold) / a.population > static_cast<double>(b.gold) / b.population;
    return a.gold * b.population > b.gold * a.population; 
}

bool compareByMedalPerCapita(const Country &a, const Country &b) {
//    return static_cast<double>(a.medal) / a.population > static_cast<double>(b.medal) / b.population;
    return a.medal * b.population > b.medal * a.population;
}

vector<int> getRankings(const vector<Country> &countries, bool(*compare)(const Country&, const Country&)) {
    vector<Country> sortedCountries = countries;
    // 根據具體規則對國家進行排序，雖然打亂了國家順序，但透過id欄位依舊可以快速定位！ 
    sort(sortedCountries.begin(), sortedCountries.end(), compare);
    // 生成相應排名（從rank=1開始，如果排名相同，繼承上一個rank； 否則採用最新的rank（rank++））  
    vector<int> rankings(countries.size());
    int rank = 1;
    for (int i = 0; i < sortedCountries.size(); ++i) {
    	// 這裡巧妙呼叫函式指標進行排名判等（如果前<=後 && 後<=前 ，那麼必然是 前 == 後） 
        if (i > 0 && !compare(sortedCountries[i - 1], sortedCountries[i]) && !compare(sortedCountries[i], sortedCountries[i - 1])) {
            rankings[sortedCountries[i].id] = rankings[sortedCountries[i - 1].id];
        } else {
            rankings[sortedCountries[i].id] = rank;
        }
        rank++;
    }
    
    return rankings;
}

int main() {
    int N, M;
    while (cin >> N >> M) {
        vector<Country> countries(N);
        for (int i = 0; i < N; ++i) {
            countries[i].id = i;
            cin >> countries[i].gold >> countries[i].medal >> countries[i].population;
        }
        
        vector<int> countryIds(M);
        for (int i = 0; i < M; ++i) {
            cin >> countryIds[i];
        }
        
        // 更新各種規則下的相應排名資訊 
        vector<int> goldRankings = getRankings(countries, compareByGold);
        vector<int> medalRankings = getRankings(countries, compareByMedal);
        vector<int> goldPerCapitaRankings = getRankings(countries, compareByGoldPerCapita);
        vector<int> medalPerCapitaRankings = getRankings(countries, compareByMedalPerCapita);
        
        for (int countryId : countryIds) {
            vector<pair<int, int>> rankings = {
                {goldRankings[countryId], 1},
                {medalRankings[countryId], 2},
                {goldPerCapitaRankings[countryId], 3},
                {medalPerCapitaRankings[countryId], 4}
            };
            sort(rankings.begin(), rankings.end()); // 對於pair型別比較，首先比較第一個元素，然後比較第二個元素 
            cout << rankings[0].first << ":" << rankings[0].second << endl;
        }
        cout << endl;
    }
    return 0;
}

相關文章