题目描述
给出 N 名运动员的成绩,找出他们的相对名次并授予前三名对应的奖牌。前三名运动员将会被分别授予 “金牌”,“银牌” 和“ 铜牌”(”Gold Medal”, “Silver Medal”, “Bronze Medal”)。
(注:分数越高的选手,排名越靠前。)
样例
输入: [5, 4, 3, 2, 1]
输出: ["Gold Medal", "Silver Medal", "Bronze Medal", "4", "5"]
解释: 前三名运动员的成绩为前三高的,因此将会分别被授予 “金牌”,“银牌”和“铜牌” ("Gold Medal", "Silver Medal" and "Bronze Medal").
余下的两名运动员,我们只需要通过他们的成绩计算将其相对名次即可。
注意
- N 是一个正整数并且不会超过 10000。
- 所有运动员的成绩都不相同。
算法
(排序) $O(n \log n)$
- 将所有运动员的分数和编号组成
pair<int, int>
,然后按照分数进行统一排序。 - 排序后,根据
pair
编号赋予奖牌或排名。
时间复杂度
- 排序的时间复杂度为 $O(n \log n)$,故总时间复杂度为 $O(n \log n)$。
C++ 代码
class Solution {
public:
vector<string> findRelativeRanks(vector<int>& nums) {
int n = nums.size();
vector<pair<int, int>> aths(n);
vector<string> ans(n);
for (int i = 0; i < n; i++)
aths[i] = make_pair(nums[i], i);
sort(aths.begin(), aths.end());
reverse(aths.begin(), aths.end());
for (int i = 0; i < n; i++) {
if (i == 0)
ans[aths[i].second] = "Gold Medal";
else if (i == 1)
ans[aths[i].second] = "Silver Medal";
else if (i == 2)
ans[aths[i].second] = "Bronze Medal";
else
ans[aths[i].second] = to_string(i + 1);
}
return ans;
}
};