首页 > itarticle > [leetcode] problem 506 – relative ranks

[leetcode] problem 506 – relative ranks

admin 11月 13, 2020 0

Given scores of N athletes, find their relative ranks and the people with the top three highest scores, who will be awarded medals: “Gold Medal”, “Silver Medal” and “Bronze Medal”.

Example

Input: [5, 4, 3, 2, 1]

Output: [“Gold Medal”, “Silver Medal”, “Bronze Medal”, “4”, “5”]

Explanation: The first three athletes got the top three highest scores, so they got “Gold Medal”, “Silver Medal” and “Bronze Medal”.
For the left two athletes, you just need to output their relative ranks according to their scores.

Note

N is a positive integer and won’t exceed 10,000.
All the scores of athletes are guaranteed to be unique.

Code

public class  {
    private int idx;
    private int score;

    public (int idx, int score) {
        this.idx = idx;
        this.score = score;
    }
}

public String[] findRelativeRanks(int[] nums) {
    int n = nums.length;
    String[] result = new String[n];
    Rank[] ranks = new Rank[n];
    String[] medals = {"Gold", "Silver", "Bronze"};

    for (int i = 0; i < n; i++)
        ranks[i] = new Rank(i, nums[i]);

    Arrays.sort(ranks, new Comparator<Rank>() {
        
        public int compare(Rank o1, Rank o2) {
            return o2.score - o1.score;
        }
    });

    for (int i = 0; i < n; i++) {
        Rank rank = ranks[i];

        if (i < 3)
            result[rank.idx] = medals[i] + " Medal";
        else
            result[rank.idx] = String.valueOf(i + 1);
    }

    return result;
}