目标
给你一个整数数组 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]
说明:
- 0 <= arr.length <= 10^5
- -10^9 <= arr[i] <= 10^9
思路
有一个整数数组 arr,返回每个元素排序后的序号,序号从 1 开始,如果元素值相同那么序号也相同,并且后面的序号无需考虑前面重复的元素,比如 [10 20 20 30],返回 [1, 2, 2, 3] 而不是 [1, 2, 2, 4]。
排序 arr 的下标数组 index,根据排序后的 index 返填原数组的序号,注意去重即可。
代码
/**
* @date 2026-07-14 10:11
*/
public class ArrayRankTransform1331 {
public int[] arrayRankTransform(int[] arr) {
int n = arr.length;
Integer[] index = new Integer[n];
Arrays.setAll(index, i -> i);
Arrays.sort(index, (a, b) -> (arr[a] - arr[b]));
int prev = Integer.MIN_VALUE;
int sort = 0;
for (int i = 0; i < n; i++) {
if (prev != arr[index[i]]){
sort++;
}
prev = arr[index[i]];
arr[index[i]] = sort;
}
return arr;
}
}
性能
