Q:
给出 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。
- 所有运动员的成绩都不相同。
思路:
首先感觉是一个排序的问题,或者是从数组中找出不同大小值所在的位置,两个提示中,N不会超过10000说明对算法的效率有要求,不能使用太多循环结构,所有运动员成绩不相同可以保证不会有重复的分数,首先想到的是对数组排序,但是那样会打乱原有数组的位置,首先使用暴力求解试一下。
class Solution:
def findRelativeRanks(self, nums):
"""
:type nums: List[int]
:rtype: List[str]
"""
temp = nums[:]
temp.sort(reverse=True)
res = [0 for _ in range(len(nums))]
for i in range(len(temp)):
if i == 0:
res[nums.index(temp[i])] = 'Gold Medal'
elif i == 1:
res[nums.index(temp[i])] = 'Silver Medal'
elif i == 2:
res[nums.index(temp[i])] = 'Bronze Medal'
else:
res[nums.index(temp[i])] = str(i+1)
return res
直接暴力求解结果,代码如上,结果肯定是不理想的,但是起码审题木有错误。
思路2: 使用字典替换 list.index() 因为字典的查询速度是O(1),因此当数组元素唯一的时候可以使用字典作为检索的方式,能够大大提升效率,新代码如下:
class Solution:
def findRelativeRanks(self, nums):
"""
:type nums: List[int]
:rtype: List[str]
"""
temp = nums[:]
temp.sort(reverse=True)
dic = {}
lens = len(temp)
dic[temp[0]] = 'Gold Medal'
if lens > 1:
dic[temp[1]] = 'Silver Medal'
if lens > 2:
dic[temp[2]] = 'Bronze Medal'
for n in range(3, lens):
dic[temp[n]] = str(n + 1)
res = [dic[n] for n in nums]
return res