leetcode506. 相对名次(每日一题)

力扣链接

题目描述

给你一个长度为 n 的整数数组 score ，其中 score[i] 是第 i 位运动员在比赛中的得分。所有得分都 互不相同 。

运动员将根据得分 决定名次 ，其中名次第 1 的运动员得分最高，名次第 2 的运动员得分第 2 高，依此类推。运动员的名次决定了他们的获奖情况：

名次第 1 的运动员获金牌 "Gold Medal" 。
名次第 2 的运动员获银牌 "Silver Medal" 。
名次第 3 的运动员获铜牌 "Bronze Medal" 。
从名次第 4 到第 n 的运动员，只能获得他们的名次编号（即，名次第 x 的运动员获得编号 "x"）。
使用长度为 n 的数组 answer 返回获奖，其中 answer[i] 是第 i 位运动员的获奖情况。

示例 1：

输入：score = [5,4,3,2,1]
输出：["Gold Medal","Silver Medal","Bronze Medal","4","5"]
解释：名次为 [1st, 2nd, 3rd, 4th, 5th] 。

示例 2：

输入：score = [10,3,8,9,4]
输出：["Gold Medal","5","Bronze Medal","Silver Medal","4"]
解释：名次为 [1st, 5th, 3rd, 2nd, 4th] 。

提示：

• n == score.length
• 1 <= n <= 104
• 0 <= score[i] <= 106
• score 中的所有值 互不相同

解题思路

官方题解链接
使用优先队列，自定义排序方式

代码

class Solution {
    
    
    public String[] findRelativeRanks(int[] score) {
    
    
        PriorityQueue<int[]> queue = new PriorityQueue<>((x, y) -> y[0] - x[0]);
        int len = score.length;
        for (int i = 0; i < len; i++) {
    
    
            queue.offer(new int[]{
    
    score[i], i});
        }

        String[] strs = new String[len];
        int count = 0;
        while (!queue.isEmpty()) {
    
    
            int[] temp = queue.poll();
            if (count == 0) {
    
    
                strs[temp[1]] = "Gold Medal";
                count++;
            } else if (count == 1) {
    
    
                strs[temp[1]] = "Silver Medal";
                count++;
            } else if (count == 2) {
    
    
                strs[temp[1]] = "Bronze Medal";
                count++;
            } else {
    
    
                count++;
                strs[temp[1]] = "" + count;
            }
        }

        return strs;
    }

}

复杂度

时间复杂度: O(nlog(n))
空间复杂度: O(n)