Leetcode #506 Relative Ranks

Description

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”.

Note

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

Example

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.

Explain

根据分数用字典映射一个排名

Code

class Solution(object):
    def findRelativeRanks(self, nums):
        """
        :type nums: List[int]
        :rtype: List[str]
        """
        rank = sorted(nums, reverse=True)
        rank_len = len(rank)
        rank_dict = dict()
        for i in range(0, rank_len):
            rank_dict[rank[i]] = i
        res = []
        for i in nums:
            if rank_dict[i] == 0:
                res.append("Gold Medal")
            elif rank_dict[i] == 1:
                res.append("Silver Medal")
            elif rank_dict[i] == 2:
                res.append("Bronze Medal")
            else:
                res.append(str(rank_dict[i]+1))
        return res