给出 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。
所有运动员的成绩都不相同。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/relative-ranks
class Solution:
def findRelativeRanks(self, nums: List[int]) -> List[str]:
n = len(nums) # 数组长度
ans = [0]*n # 结果
dt = {} # 哈希, 记录每个元素开始所在的位置
for i in range(n):
dt[nums[i]] = i
#nums.sort()
self.QSort(nums,0,n-1) # 快速排序
if n==0:
return None
for i in range(n):
if i == 0:
ans[dt[nums[i]]] = "Gold Medal"
elif i == 1:
ans[dt[nums[i]]] = "Silver Medal"
elif i == 2:
ans[dt[nums[i]]] = "Bronze Medal"
else:
ans[dt[nums[i]]] = str((i+1)) # 字符串
return ans
def QSort(self,nums, low, hight):
if low < hight:
ans = self.QSpass(nums,low,hight) # 关键部分,一趟快速排序
self.QSort(nums,low,ans-1) # 左边的子表
self.QSort(nums,ans+1,hight) # 右边的子表
def QSpass(self, nums, low, hight):
'''把大于key的放在左子表,小的放在右子表'''
key = nums[low] # 以第一位作为关键字
while low < hight:
# 从hight往low走,找到大于key的,就交换
while low<hight and nums[hight] <= key:
hight -= 1
if low < hight:
#self.swap(nums, low, hight)
nums[low] = nums[hight]
low += 1
# 从low往hight走,遇到比key小的,和hight交换
while low < hight and nums[low] >= key:
low+=1
if low<hight:
#self.swap(nums, low,hight)
nums[hight] = nums[low]
hight -= 1
nums[low] = key # 把key放在low==hight的位置
return low # 返回 key,关键字的小标