这是我参与更文挑战的第22天，活动详情查看：更文挑战

1122. 数组的相对排序

题目描述

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

提示：

1 <= arr1.length, arr2.length <= 1000
0 <= arr1[i], arr2[i] <= 1000
arr2 中的元素 arr2[i] 各不相同
arr2 中的每个元素 arr2[i] 都出现在 arr1 中

友情链接

leetcode-cn.com/problems/re…

思路解析

因为题目已经说明了数组的值在 0~1000 范围内。可以用桶排序算法来解这道题。

一开始就新建一个长度为 1001 的数组作为空桶，和一个专门存储最终结果的数组，长度等于 arr1 的长度。
顺序遍历 arr1 数组，初始化桶数组。数组元素值作为数组桶的位置，数组桶的元素值就是 arr1 数组元素出现的次数。
顺序遍历 arr2 ，先将桶里面等于的 arr2 的元素设置进结果数组中。
最后从头开始遍历元素大于 0 的桶元素，按计数设置进结果数组中。
结果数组即为所求。

算法实现

class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        int[] bucket=new int[1001],ret=new int[arr1.length];
        for(int a1:arr1){
            bucket[a1]++;
        }

        int index=0;
        for(int a2:arr2){
            for(int i=0;i<bucket[a2];i++){
                ret[index++]=a2;
            }
            bucket[a2]=0;
        }

        for(int i=0;i<1001;i++){
            if(bucket[i]>0){
                while(bucket[i]-->0){
                    ret[index++]=i;
                }
            }
        }
        return ret;
    }
}

运行结果

可以看到内存占用了 38.4MB 。从这里也能明显感觉到桶排序属于典型的用空间换时间的算法。

kylin来刷题啦-Day11：1122. 数组的相对排序