LeetCode 506. 相对名次（python）

题目链接

题目描述：

给出 N 名运动员的成绩，找出他们的相对名次并授予前三名对应的奖牌。前三名运动员将会被分别授予 “金牌”，“银牌” 和“ 铜牌”（“Gold Medal”, “Silver Medal”, “Bronze Medal”）。

(注：分数越高的选手，排名越靠前。)

示例 1:

输入: [5, 4, 3, 2, 1]
输出: [“Gold Medal”, “Silver Medal”, “Bronze Medal”, “4”, “5”]
解释: 前三名运动员的成绩为前三高的，因此将会分别被授予 “金牌”，“银牌”和“铜牌” (“Gold Medal”, “Silver Medal” and “Bronze Medal”).
余下的两名运动员，我们只需要通过他们的成绩计算将其相对名次即可。
提示:

N 是一个正整数并且不会超过 10000。
所有运动员的成绩都不相同。

解题思路：

用nn表示从大到小排序后nums。
对于nums中的每个元素，其排名等于该元素在nn中的索引加1
再将前三名换成相应的奖牌即可。
但是这种方法用时比较长，应该是每次在查找值在nn中的索引中都会遍历一遍列表

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

评论区答案：

        A = []
        for i in range(len(nums)):
            A.append([i, nums[i]])
        print(A)
        A.sort(key = lambda x: x[1], reverse = True)#按值从大到小排序
        print(A)
        if len(nums) >= 1:
            nums[A[0][0]] = "Gold Medal"
        if len(nums) >= 2:
            nums[A[1][0]] = "Silver Medal"
        if len(nums) >= 3:
            nums[A[2][0]] = "Bronze Medal"
        for i in range(3, len(A)):
            nums[A[i][0]] = str(i + 1)
        return nums

用A存放nums的索引和值，然后将值按照从大到小排序
这样排序后的A的索引+1（i+1）是其排名
每一个A的第一个值是其原来的索引：nums[A[i][0]] = str(i + 1)也就是说，将原位置处的值替换为排名