【一次过】Lintcode 1200. 相对排名

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/majichen95/article/details/83651462

根据N名运动员的得分，找到他们的相对等级和获得最高分前三名的人，他们将获得奖牌：“金牌”，“银牌”和“铜牌”。

样例

例子 1:

输入: [5, 4, 3, 2, 1]
输出: ["Gold Medal", "Silver Medal", "Bronze Medal", "4", "5"]
说明：前三名运动员获得前三名最高分，因此获得“金牌”，“银牌”和“铜牌”。
对于右边两名运动员，你只需要根据他们的分数输出他们的相对等级。

注意事项

N是正整数，并且不超过10,000。
所有运动员的成绩都保证是独一无二的。

---

解题思路：

需要排序，就用TreeMap结构，key存储数组中的值，value存储对应的下标，TreeMap默认按key升序排列。

public class Solution {
    /**
     * @param nums: List[int]
     * @return: return List[str]
     */
    public String[] findRelativeRanks(int[] nums) {
        // write your code here
        TreeMap<Integer,Integer> map = new TreeMap<>();//key存储数组中的值，value存储对应的下标，TreeMap默认按key升序排列
        for(int i=0; i<nums.length; i++)
            map.put(nums[i], i);
        
        String[] res = new String[nums.length];
        int rank = map.size();
        
        for(Integer i : map.keySet()){
            if(rank == 1)
                res[map.get(i)] = "Gold Medal";
            else if(rank == 2)
                res[map.get(i)] = "Silver Medal";
            else if(rank == 3)
                res[map.get(i)] = "Bronze Medal";
            else
                res[map.get(i)] = "" + rank;
                
            rank--;
        }
        
        return res;
    }
}