LeetCode1122. 数组的相对排序（Java）

题目：

给你两个数组，arr1 和 arr2，

• arr2 中的元素各不相同

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

• 解法一

/*
思路：

利用类似计数排序的思路
	先计数排序（计数排序清查看 csdn——数据结构与算法——排序）
	再使用arr2中元素对排序的数组进行微调
*/
class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        int[] count=new int[1001];
        for(int i:arr1){
            count[i]++;
        }
        int start=0;
        for(int i:arr2){
            while(count[i]>0){
                arr1[start++]=i;
                count[i]--;
            }
        }
        for(int i=0;i<1001;i++){
            while(count[i]>0){
                arr1[start++]=i;
                count[i]--;
            }
        }
        return arr1;
    }
}

• 解法二

/*
思路：

1)先将 arr2 中的元素存入到一个 Map 中，值为 key，下标索引为 index，作为排序的标准
2)将 arr1 分为两个部分，第一个部分包含 arr2 的元素，第二个部分不包含 arr2 的元素
3)将第一个部分按照 Map 中的规则进行排序，第二个部分按照升序排序
4)将两个部分的数组拼接起来
*/
class Solution {
    public int[] relativeSortArray(int[] arr1, int[] arr2) {
        HashMap<Integer, List<Integer>> map = new HashMap<>();
        List<Integer> list = new ArrayList<>();
        int[] res = new int[arr1.length];
        int idx = 0;
        for (int i : arr2
                ) {
            map.put(i, new ArrayList<Integer>());
        }
        for (int i :
                arr1) {
            if (map.containsKey(i)) {
                map.get(i).add(i);
            } else {
                list.add(i);
            }
        }
        for (int a : arr2
                ) {
            List<Integer> list1 = map.get(a);
            for (int b : list1
                    ) {
                res[idx++] = b;
            }
        }
        Collections.sort(list);

        for (int b : list
                ) {
            res[idx++] = b;
        }
        return res;
    }
}