golang 数组的相对排序

最新推荐文章于 2022-07-26 14:56:13 发布

pardon110

最新推荐文章于 2022-07-26 14:56:13 发布

阅读量721

点赞数 2

分类专栏： Golang 探究算法

本文链接：https://blog.csdn.net/u011584949/article/details/109688110

版权

算法同时被 2 个专栏收录

63 篇文章 0 订阅

订阅专栏

Golang 探究

45 篇文章 0 订阅

订阅专栏

该博客介绍了如何在Go语言中根据特定规则对数组进行排序。内容涉及自定义排序函数，通过创建一个映射记录元素在参考数组中的相对位置，然后基于这个映射进行排序。两种不同的实现方式被提出，一种利用`sort.Slice`直接排序，另一种通过构建新的数组结构再进行排序。这两种方法都能确保arr1中元素的相对顺序与arr2相同，未在arr2中出现的元素会按升序排列。

摘要由CSDN通过智能技术生成

题面

两个数组，arr1 和 arr2，

arr2 中的元素各不相同
arr2 中的每个元素都出现在 arr1 中
对 arr1 中的元素进行排序，使 arr1 中项的相对顺序和 arr2 中的相对顺序相同。未在 arr2 中出现过的元素需要按照升序放在 arr1 的末尾。

输入：arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6]
输出：[2,2,2,1,4,3,3,9,6,7,19]

分析

自定义排序，合理描述规则
排序规则与索引，值，可哈希相关

自定义规则

import "sort"
func relativeSortArray(arr1 []int, arr2 []int) []int {
    rank := map[int]int{}
    // 记录相对位置
    for i, v := range arr2 {
        rank[v] = i
    }
    // 自定义排序
    sort.Slice(arr1, func(i, j int) bool {
        x, y := arr1[i], arr1[j]
        rankX, hasX := rank[x]
        rankY, hasY := rank[y]
        if hasX && hasY {
            return rankX < rankY
        }
        if hasX || hasY {
            return hasX
        }
        return x < y
    })
    return arr1
}

合规建数组

import "sort"
func relativeSortArray(arr1 []int, arr2 []int) []int {
    m := make(map[int][]int,0)
    for i:=0;i<len(arr2);i++{
        m[arr2[i]] = []int{}
    }
    rst :=[]int{}
    for i:=0;i<len(arr1);i++ {
        v := arr1[i]
        if _,ok := m[v];ok {
            m[v] = append(m[v], v)
        }else{
            rst = append(rst, v)
        }
    }
    src :=[]int{}
    for _, v := range arr2 {
        src = append(src, m[v]...)
    }
    sort.Ints(rst)
    return append(src, rst...)
}