golang 数组的相对排序

题面

两个数组，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...)
}