1. 程式人生 > >【一次過】Lintcode 1200. 相對排名

【一次過】Lintcode 1200. 相對排名

根據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;
    }
}