数组的相对排序

给你两个数组，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]

  • 输入：arr1 = [28,6,22,8,44,17], arr2 = [22,28,8,6]

  • 输出：[22,28,8,6,17,44]

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
var relativeSortArray = function(arr1, arr2) {
    // 转换成次数表达的数组
    // arr1 === [1,1,1,2,2,3]
    // repo === [,3,2,1]
    const repo = []
    arr1.forEach(num => {
        if (repo[num]) {
            repo[num]++
        } else {
            repo[num] = 1
        }
    })
    //  循环arr2拿到排序，然后把统计出来repo的次数兑掉
    const result = []
    arr2.forEach(num => {
        while (repo[num]) {
            result.push(num)
            repo[num]--
        }
    })
    // 循环剩下的repo，依次推进结果
    for (let i = 0; i < repo.length; i++) {
        while(repo[i]) {
            result.push(i)
            repo[i]--
        }
    }
    return result
};