【算法与数据结构相关】【LeetCode】【506 相对名次】【Python】

题目:给出 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").
余下的两名运动员,我们只需要通过他们的成绩计算将其相对名次即可。

提示:

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

思路:这道题倒是不难,但是自己写的代码太罗嗦了

class Solution(object):
    def findRelativeRanks(self, nums):
        """
        :type nums: List[int]
        :rtype: List[str]
        """
        nums_sort = sorted(nums, reverse=True)
        result = [-1]*len(nums)
        medal = ["Gold Medal", "Silver Medal", "Bronze Medal"]
        if len(nums) <=2:
            for i in range(len(nums_sort)):
                result[nums.index(nums_sort[i])] = medal[i]
        elif len(nums)>2:
            result[nums.index(nums_sort[0])] = medal[0]
            result[nums.index(nums_sort[1])] = medal[1]
            result[nums.index(nums_sort[2])] = medal[2]
            for index, item in enumerate(nums_sort[3:]):
                result[nums.index(item)] = str(index+4)
        return result

看了排名前几的代码,及其精简,自己的Python用的还是太肤浅了:

class Solution(object):
    def findRelativeRanks(self, nums):
        """
        :type nums: List[int]
        :rtype: List[str]
        """
        sort = sorted(nums)[::-1]
        rank = ["Gold Medal", "Silver Medal", "Bronze Medal"] + map(str, range(4, len(nums) + 1))
        return map(dict(zip(sort, rank)).get, nums)

 


版权声明:本文为gq930901原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。