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 中

解题思路

  1. #define N 1010
  2. class Solution {
  3. public:
  4.     vector<int> relativeSortArray(vector<int>& arr1, vector<int>& arr2) {
  5.         vector<int> hash(N,0);
  6.         int len1 = arr1.size();
  7.         int len2 = arr2.size();
  8.         vector<int> res;
  9.         for(int i=0;i < len1;i++) {
  10.             hash[arr1[i]]++;
  11.         }
  12.         for(int i=0;i < len2;i++) {
  13.             for(int j=0;j<hash[arr2[i]];j++)
  14.                 res.push_back(arr2[i]);
  15.             hash[arr2[i]] = 0;
  16.         }
  17.         for(int i=0;i<N;i++) {
  18.              for(int j=0;j<hash[i];j++)
  19.                 res.push_back(i);
  20.         }
  21.         return res;
  22.     }
  23. };