题目描述:
给出 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;
}
}