相对名次（LeetCode）-优快云博客

本文介绍了一种算法，用于确定运动员的相对名次并分配前三名的奖牌。该算法使用TreeMap对成绩进行排序，然后将前三名的成绩对应修改为“金牌”，“银牌”和“铜牌”。其余名次则根据成绩计算得出。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

给出 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。
所有运动员的成绩都不相同。

解题思路：

首先将所有的成绩作为key，存放进有序的TreeMap中。
将TreeMap中key最大的前三个数据的value值依次改为"Gold Medal"，"Silver Medal" ， "Bronze Medal"。其他数据的value值改成其key所对应的次序（按照降序计算）。
创建一个字符串数组，大小与给定数组相同。
从前往后遍历给定数组，以给定数组的每个数组作为key，找到map中对应的value，并将value，存到字符串数组中。
返回字符串数组

代码实现：

class Solution {
    public String[] findRelativeRanks(int[] nums) {
        if (nums == null || nums.length < 1) {
            return new String[]{};
        }
        TreeMap<Integer,String> map = new TreeMap<>();
        for (int i = 0; i < nums.length; i++) {
            map.put(nums[i],"");
        }
        int flag = 1;
        int size = map.size();
        Iterator<Integer> iterator = map.keySet().iterator();
        while (iterator.hasNext()) {
            int key = iterator.next();
            if (flag == size - 2) {
                map.put(key,"Bronze Medal");
            } else if (flag == size - 1) {
                map.put(key,"Silver Medal");
            } else if (flag == size) {
                map.put(key,"Gold Medal");
            } else {
                map.put(key,String.valueOf(size- flag + 1));
            }
            flag++;
        }
        String[] arr = new String[nums.length];
        for (int i = 0; i < nums.length; i++) {
            arr[i] = map.get(nums[i]);
        }
        return arr;
    }
}