题目:给出 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").
余下的两名运动员,我们只需要通过他们的成绩计算将其相对名次即可。提示:
- N 是一个正整数并且不会超过 10000。
- 所有运动员的成绩都不相同。
思路:这道题倒是不难,但是自己写的代码太罗嗦了
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版权协议,转载请附上原文出处链接和本声明。