0
点赞
收藏
分享

微信扫一扫

算法记录(python,go)

萨科潘 2022-02-16 阅读 84

一、数组序号转换

给你一个整数数组 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]

解法1:python

class Solution:
    def arrayRankTransform(self, arr: List[int]) -> List[int]:
        end_ls = {}
        for i, v in enumerate(sorted(set(arr)), 1):
            end_ls[v] = i
        return [end_ls[v] for v in arr]

解法2:goland

// 缺点,内存消耗大
func arrayRankTransform(arr []int) []int {
    arr1 := make([]int, len(arr))
	copy(arr1, arr)
	sort.Ints(arr)
	// 去重, 获得排序的map
	flag_map := make(map[int]int)
	flag := 1
	for _, v := range arr {
		if _, ok := flag_map[v]; !ok {
			flag_map[v] = flag
			flag++
		}
	}

	end_ls := []int{}
	for _, v := range arr1 {
		end_ls = append(end_ls, flag_map[v])
	}
	return end_ls
}

后续会继续将leetcode上做的题目,记录到此处

举报

相关推荐

0 条评论