LeetCode.1122 数组的相对排序

118 阅读1分钟

「这是我参与2022首次更文挑战的第22天,活动详情查看:2022首次更文挑战」。

题目描述:

1122. 数组的相对排序 - 力扣(LeetCode) (leetcode-cn.com)

给你两个数组,arr1 和 arr2arr2 中的元素各不相同,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 <= arr1.length, arr2.length <= 1000
  • 0 <= arr1[i], arr2[i] <= 1000
  • arr2 中的元素 arr2[i]  各不相同 
  • arr2 中的每个元素 arr2[i] 都出现在 arr1 中

思路分析

计数排序

注意到本题中元素的范围为 [0, 1000][0,1000],这个范围不是很大,我们也可以考虑不基于比较的排序,例如「计数排序」。

具体步骤如下:

1、定义一个数组 hash,长度为 1001,用于模拟哈希表。

2、将 arr1 中的所有元素存放在 hash 数组中,其中的键为 arr1 中的元素,值为 arr1 中的元素出现的次数;

3、定义索引 i,初始化为 0,用于重置数组 arr1 中的元素值。

4、遍历数组 arr2,只要 arr2 中的元素在数组 hash 中存在,则将其赋值给 arr1,并对该元素在 hash 中出现的频次减一;

5、针对不是数组 arr2 中的元素,遍历整个数组 hash,只要其元素出现的次数在一次及以上,将其赋值给 arr1,并将该元素在 hash 中出现的频次减一。

AC代码

class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
            int i = 0;
            int [] hash = new int [1001];
            for (int n : arr1) {
                hash[n]++;
            }

            for (int n : arr2) {
                while (hash[n] > 0) {
                    arr1[i++] = n;
                    hash[n]--;
                }
            }

            for (int n = 0; n < hash.length; ++n) {
                while (hash[n] > 0) {
                    arr1[i++] = n;
                    hash[n]--;
                }
            }

            return arr1;
    }
}

参考

数组的相对排序 - 数组的相对排序 - 力扣(LeetCode) (leetcode-cn.com)

1122. 数组的相对排序 - 数组的相对排序 - 力扣(LeetCode) (leetcode-cn.com)