问题描述:
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:
- N is a positive integer and won't exceed 10,000.
- All the scores of athletes are guaranteed to be unique.
我的代码:
class Solution(object):
def findRelativeRanks(self, nums):
"""
:type nums: List[int]
:rtype: List[str]
"""
temp = sorted(nums,reverse=True)
r = []
for n in nums:
no = temp.index(n)+1
if no == 1:
no = "Gold Medal"
elif no == 2:
no = "Silver Medal"
elif no == 3:
no = "Bronze Medal"
r.append(str(no))
return r
3行python:
def findRelativeRanks(self, nums):
sort = sorted(nums)[::-1] #可以用reverse=True代替,作者说两者复杂度是一样的,大师[::-1]只要打6个字符.....大牛就是大牛,任性
#这句是创建一个形如这样的list ["Gold Medal", "Silver Medal", "Bronze Medal", "4","5",..."n"]
rank = ["Gold Medal", "Silver Medal", "Bronze Medal"] + map(str, range(4, len(nums) + 1))
#这句return最能体现python强大的库函数,zip吧两个list一一对应的映射起来形成dict,在把nums里面的元素全部在dict里get一遍
return map(dict(zip(sort, rank)).get, nums)