LEETCODE 1122. 数组的相对排序

class Solution {
public:
    vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2){
        //计数排序
        vector<int> map(1001);
        for(int i=0;i<arr1.size();i++){
            map[arr1[i]]+=1;     
        }
        int j=0;
        for(int i=0;i<arr2.size();i++){
            while(map[arr2[i]]--){
                arr1[j++]=arr2[i];
            }
        }
        // int t=i;
        for(int i=0;i<map.size();i++){
            while(map[i]>0){
                arr1[j++]=i;
                map[i]-=1;
            }
        }
        return arr1;


    }
};