leetcode之哈希表二:数组的相对排序

原题:

力扣https://leetcode-cn.com/problems/relative-sort-array/

一、题目要求

给你两个数组，arr1 和 arr2，

arr2 中的元素各不相同
arr2 中的每个元素都出现在 arr1 中
对 arr1 中的元素进行排序，使 arr1 中项的相对顺序和 arr2 中的相对顺序相同。未在 arr2 中出现过的元素需要按照升序放在 arr1 的末尾。

示例：

提示：

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

二、解题

package com.leetcode.hash;

import java.util.*;

public class Solution13 {
    public static void main(String[] args) {
        int[] arr1 = {2,21,43,38,0,42,33,7,24,13,12,27,12,24,5,23,29,48,30,31}, arr2 = {2,42,38,0,43,21};
        System.out.println(Arrays.toString(relativeSortArray(arr1, arr2)));
    }

    public static int[] relativeSortArray(int[] arr1, int[] arr2) {
        HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
        TreeMap<Integer, Integer> treeMap = new TreeMap<Integer, Integer>();

        for (int i:arr2) {
            map.put(i, 0);
        }

        for (int i:arr1) {
            if (map.containsKey(i)) {
                map.put(i, map.get(i) + 1);
            } else {
                if (treeMap.containsKey(i)) {
                    treeMap.put(i, treeMap.get(i) + 1);
                } else {
                    treeMap.put(i, 1);
                }
            }
        }

        ArrayList<Integer> list = new ArrayList<Integer>();
        for (int i:arr2) {
            int count = map.get(i);
            for (int j = 0; j < count; j++) {
                list.add(i);
            }
        }

        for (int i:treeMap.keySet()) {
            int count = treeMap.get(i);
            for (int j = 0; j < count; j++) {
                list.add(i);
            }
        }

        int[] arr = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            arr[i] = list.get(i);
        }
        return arr;
    }
}

三、运行结果

四、提交结果