1331. 数组序号转换

数组序号转换

题目描述:

给你一个整数数组 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 <= 105
-109 <= arr[i] <= 109

思路:

时间复杂度:O(nlogn),空间复杂度O(n)

使用排序,然后利用map来存位置

代码:


func arrayRankTransform(arr []int) []int {
    n := len(arr)
    arr2 := make([]int, n)
    copy(arr2,arr)
    sort.Ints(arr2)
    mp := make(map[int]int,0)
    cnt := 1
    for i:=0;i<n;i++{
        if i!= 0 && arr2[i] == arr2[i-1] {
            continue
        }
        mp[arr2[i]] = cnt
        cnt ++
    }
    var res []int
    for i:=0;i<n;i++{
        res = append(res,mp[arr[i]])
    }
    return res
}

代码效率:

执行用时:64 ms, 在所有 Go 提交中击败了79.17%的用户
内存消耗:14.2 MB, 在所有 Go 提交中击败了8.33%