506. Relative Ranks

来源:互联网 发布:会计行业 知乎 编辑:程序博客网 时间:2024/06/07 07:56

题目:

Given scores of N athletes, find their relative ranks and the people with the top three highest scores, who will be awarded medals: "Gold Medal", "Silver Medal" and "Bronze Medal".

Example 1:

Input: [5, 4, 3, 2, 1]Output: ["Gold Medal", "Silver Medal", "Bronze Medal", "4", "5"]Explanation: The first three athletes got the top three highest scores, so they got "Gold Medal", "Silver Medal" and "Bronze Medal". 
For the left two athletes, you just need to output their relative ranks according to their scores.

Note:

  1. N is a positive integer and won't exceed 10,000.
  2. All the scores of athletes are guaranteed to be unique.

思路:

本题需要活用标准容器:pair及priority_queue,先维护pair数值对,统计nums[i]及对应下标,然后根据大顶堆来存储数据,最后再赋值(to_string)

代码:

class Solution {public:    vector<string> findRelativeRanks(vector<int>& nums) {        priority_queue<pair<int,int> > pq;        for(int i=0;i<nums.size();i++)        {            pq.push(make_pair(nums[i],i));        }        vector<string> res(nums.size(),"");        int count = 1;        for(int i=0; i<nums.size();i++)        {            if(count==1) {res[pq.top().second] = "Gold Medal"; count++;}            else if(count==2) {res[pq.top().second] = "Silver Medal"; count++;}            else if(count==3) {res[pq.top().second] = "Bronze Medal"; count++;}            else {res[pq.top().second] = to_string(count); count++;}            pq.pop();        }        return res;    }};


原创粉丝点击