LeetCode-Relative Sort Array

最新推荐文章于 2023-02-25 23:03:36 发布

原创最新推荐文章于 2023-02-25 23:03:36 发布 · 510 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #Java #Sort

LeetCode 专栏收录该内容

306 篇文章

订阅专栏

博客围绕对数组arr1排序展开，要求按arr2元素顺序排，未在arr2出现的元素升序放尾部。给出解法，先利用哈希表存arr1元素出现次数，遍历arr2插入对应元素到结果数组，再插入未出现元素到尾部，最后对尾部排序，使用Java实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Description:
Given two arrays arr1 and arr2, the elements of arr2 are distinct, and all elements in arr2 are also in arr1.

Sort the elements of arr1 such that the relative ordering of items in arr1 are the same as in arr2. Elements that don’t appear in arr2 should be placed at the end of arr1 in ascending order.

Example 1:

Input: arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6]
Output: [2,2,2,1,4,3,3,9,6,7,19]

Constraints:

arr1.length, arr2.length <= 1000
0 <= arr1[i], arr2[i] <= 1000
Each arr2[i] is distinct.
Each arr2[i] is in arr1.

题意：对数组 $a r r 1$ 进行排序，要求首先按照 $a r r 2$ 中元素出现的顺序排序，最后将没有出现在 $a r r 2$ 中的元素按照升序排序排列在数组 $a r r 1$ 的尾部；

解法：首先，要按照 $a r r 2$ 中的顺序排序，需要找出 $a r r 1$ 中出现的所有这些元素；因此，第一步，利用哈希表存储 $a r r 1$ 中的所有元素出现的次数后，遍历 $a r r 2$ 中的元素 $k e y$ ，将 $a r r 1$ 中出现的所有 $k e y$ 插入到结果数组中；第二步，将没有在 $a r r 2$ 中出现的元素插入到结果数组的尾部；最后，对结果数组尾部那部分数组进行排序；

Java

class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int arr: arr1) {
            map.put(arr, map.getOrDefault(arr, 0) + 1);
        }
        int[] result = new int[arr1.length];
        int ind = 0;
        for (int i = 0; i < arr2.length; i++) {
            for (int j = 0; j < map.get(arr2[i]); j++) {
                result[ind++] = arr2[i];
            }
            map.remove(arr2[i]);
        }
        int st = ind;
        for (int k: map.keySet()) {
            for (int i = 0; i < map.get(k); i++) {
                result[ind++] = k;
            }
        }
        Arrays.sort(result, st, result.length);
        return result;
    }
}