【数组序号转换(1331-java)】

数组序号转换(1331-java)

给你一个整数数组 arr ，请你将数组中的每个元素替换为它们排序后的序号。

序号代表了一个元素有多大。序号编号的规则如下：

序号从 1 开始编号。
一个元素越大，那么序号越大。如果两个元素相等，那么它们的序号相同。
每个数字的序号都应该尽可能地小。
 

示例 1：

输入：arr = [40,10,20,30]
输出：[4,1,2,3]
解释：40 是最大的元素。 10 是最小的元素。 20 是第二小的数字。 30 是第三小的数字。
示例 2：

输入：arr = [100,100,100]
输出：[1,1,1]
解释：所有元素有相同的序号。
示例 3：

输入：arr = [37,12,28,9,100,56,80,5,12]
输出：[5,3,4,2,8,6,7,1,3]

public class LC238_1331_arrayRankTransform {
    //排序
    public static int[] arrayRankTransform(int[] arr) {
        //使用优先队列解决
        //1.初始化堆
        //2.填入数据
        //3.遍历
        //4.填入结果集
        if (arr == null || arr.length == 0) {
            return new int[0];
        }
        PriorityQueue<Integer> queue = new PriorityQueue<>(arr.length, Comparator.comparingInt(a -> arr[a]));
        for (int i = 0; i < arr.length; i++) {
            queue.offer(i);
        }
        int[] res = new int[arr.length];
        int rank = 1;
        int last = -1;
        while (!queue.isEmpty()) {
            Integer curr = queue.poll();
            if (last != -1 && arr[last] == arr[curr]) {
                res[curr] = res[last];
            } else {
                res[curr] = rank;
                rank++;
            }
            last = curr;
        }
        return res;
    }
    public static void main(String[] args) {
        int[] ints = arrayRankTransform(new int[]{40, 10, 20, 30});
        for (int s : ints) {
            System.out.println(s);
        }
    }
}