1122. Relative Sort Array

问题：

给定数组arr2，所含元素唯一，

给定数组arr1，对arr1排序，使得arr1的元素相对顺序为arr2的顺序，不存在于arr2中的元素，按照升序排列。

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.

　　

解法：

由于限制：所有元素大小：0 <= arr1[i], arr2[i] <= 1000

因此可以考虑使用桶排序。

计数arr1中每个元素出现的次数ncout[]，

首先，根据arr2的元素，遍历，输出ncout[a2]个a2，

同时更新ncout[a2]=0，

再次，遍历ncout，输出计数不为0的元素。

代码参考：

class Solution {
public:
    vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) {
        vector<int> res;
        int ncout[1001]={0};
        for(int a1:arr1){
            ncout[a1]++;
        }
        for(int a2:arr2){
            if(ncout[a2]>0){
                vector<int> tmp(ncout[a2], a2);
                res.insert(res.end(), tmp.begin(), tmp.end());
                ncout[a2]=0;
            }
        }
        for(int i=0; i<1001; i++){
            if(ncout[i]>0){
                vector<int> tmp(ncout[i], i);
                res.insert(res.end(), tmp.begin(), tmp.end());
            }
        }
        return res;
    }
};

另，不使用vector合并insert方法，使用push_back的时候：

但这时花费memory大于上面的合并vector方法。

class Solution {
public:
    vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) {
        vector<int> res;
        int ncout[1001]={0};
        for(int a1:arr1){
            ncout[a1]++;
        }
        for(int a2:arr2){
            while(ncout[a2]>0){
                res.push_back(a2);
                ncout[a2]--;
            }
        }
        for(int i=0; i<1001; i++){
            while(ncout[i]>0){
                res.push_back(i);
                ncout[i]--;
            }
        }
        return res;
    }
};