1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21

from typing import List


class Solution:
    def relativeSortArray(self, arr1: List[int], arr2: List[int]) -> List[int]:
        order_dict = {v: 0 for v in arr2}
        not_order_dict = {}
        result = []
        for v in arr1:
            if v in order_dict:
                order_dict[v] += 1
            else:
                not_order_dict.setdefault(v, 0)
                not_order_dict[v] += 1
        for v in arr2:
            sub_result = [v] * order_dict[v]
            result.extend(sub_result)
        for v in sorted(not_order_dict):
            sub_result = [v] * not_order_dict[v]
            result.extend(sub_result)
        return result