Every day a leetcode
题目来源:506. 相对名次
解法1:STL vector 自定义排序
使用 vector<pair<int, int>> 数组v记录原来 score 数组的下标和成绩。
对v按成绩进行降序排序。
新建一个 vector<string> 数组 ans,遍历v,按要求对 ans[v[i].first] 进行赋值:
- 若i == 0,说明成绩最高,赋值 “Gold Medal”;
- 若i == 1,说明成绩第2,赋值 “Silver Medal”;
- 若i == 2,说明成绩第3,赋值 “Bronze Medal”;
- 否则,赋值 to_string(i+1)。
注意:使用sort函数的时候因为自己要定义compare,但是会报错error: reference to non-static member function must be called。
解决方法:compare函数前面加个static就可。
代码:
/*
* @lc app=leetcode.cn id=506 lang=cpp
*
* [506] 相对名次
*/
// @lc code=start
class Solution
{
public:
// cmp函数前面加个static,否则报错:
// reference to non-static member function must be called
static bool cmp(const pair<int, int> A, const pair<int, int> B)
{
return A.second > B.second; // 降序排列
}
vector<string> findRelativeRanks(vector<int> &score)
{
int n = score.size();
vector<pair<int, int>> v;
for (int i = 0; i < n; i++)
v.push_back(pair<int, int>(i, score[i]));
sort(v.begin(), v.end(), cmp);
vector<string> ans(n);
for (int i = 0; i < n; i++)
{
if (i == 0)
ans[v[i].first] = "Gold Medal";
else if (i == 1)
ans[v[i].first] = "Silver Medal";
else if (i == 2)
ans[v[i].first] = "Bronze Medal";
else
ans[v[i].first] = to_string(i + 1);
}
return ans;
}
};
// @lc code=end
结果:
复杂度分析:
时间复杂度:O((nlogn),其中 n 为数组的长度。我们需要对数组进行一次排序,因此时间复杂度为 O(nlogn)。
空间复杂度:O(n),其中 n 为数组的长度。