【07.28 每日一题】1331. 数组序号转换

105 阅读1分钟

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]
解释: 所有元素有相同的序号。

思路

  • 简单题,采用排序和map相结合的方法
  • 首先定义一个新数组temp,深拷贝数组arr,并对其排序,将temp[i]作为key,索引index为value,最后遍历arr输出

解法

func arrayRankTransform(arr []int) []int {
    temp:=make([]int,len(arr))
    copy(temp,arr)
    sort.Ints(temp)
    dic:=map[int]int{}
    index:=1
    for i:=0;i<len(temp);i++{
        if _,ok:=dic[temp[i]];!ok{
            dic[temp[i]]=index
            index++
        }
    }
    res:=[]int{}
    for i:=0;i<len(arr);i++{
        res=append(res,dic[arr[i]])
    }
    return res
}

结果

image.png