LeetCode 1122——数组的相对排序（计数排序）

一、题目介绍

给你两个数组，arr1 和 arr2，

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

示例：

输入：arr1 = [2,3,1,3,2,4,6,7,9,2,19], arr2 = [2,1,4,3,9,6]
输出：[2,2,2,1,4,3,3,9,6,7,19]
 

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/relative-sort-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、解题思路

解决本题，可以通过以下几个步骤来完成：

（1）遍历数组arr1 ，记录每个在arr2中存在的元素及其出现次数。并将在arr2中不存在的元素保存到一个临时数组vec中。

（2）将vec中的元素按照升序进行排序。

（3）遍历arr2将arr1中相同于arr2的元素，按照arr2的顺序进行排序，存放到输入数组res中。

（4）在输出数组res的末尾，添加vec中的所有元素。

三、解题代码

class Solution {
public:
    vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) {
        if(arr1.empty() || arr2.empty())
            return vector<int>();
        vector<int> res;
        map<int, int> m1;
        set<int> st(arr2.begin(), arr2.end()); //将数组2映射到map中
        vector<int> vec;
        for(int i = 0; i < arr1.size(); ++i)
        {
            if(st.count(arr1[i]) == 0)  //如果当前元素在数组2中不存在，则保存到一个单独数组中
                vec.push_back(arr1[i]);
            else
            {
                m1[arr1[i]]++;
            }
        }
        sort(vec.begin(), vec.end()); //数组2中未出现的元素按照升序排列
        //遍历数组2，将数组1中的相同于数组2的元素，按照数组2的顺序进行排序
        for(int i = 0; i < arr2.size(); ++i)
        {
            while(m1[arr2[i]]--)
            {
                res.push_back(arr2[i]);
            }
        }
        //将数组1中的不同于数组2的元素，按照升序排列放到数组末端
        for(int i = 0; i < vec.size(); ++i)
        {
            res.push_back(vec[i]);
        }
        return res;
    }
};

四、解题结果